雅虎面试题─有两个双向循环链表A,B,知道其头指针为:pHeadA,pHeadB,请写一函数将两链表中data值相同的结点删除

来源:互联网 发布:你初到大城市知乎 编辑:程序博客网 时间:2024/05/22 05:33

有双向循环链表结点定义为:

struct node

  int data;
  struct node *front,*next;
};

有两个双向循环链表A,B,知道其头指针为:pHeadA,pHeadB,请写一函数将两链表中data值相同的结点删除。

用两个向量A_vec、B_vec分别存储链表A、B的元素值,将A_vecB_vec排序,用类似归并排序的方法把A_vecB_vec中值相同的元素放到向量common_vec中。分别遍历链表A、B,用二分查找法查看每个节点元素是否在common_vec中。

// 链表结点template <typename T>class dc_list_node{public:    dc_list_node(const T &item = T(), dc_list_node<T> *p_front = NULL, dc_list_node<T> *p_next = NULL):        m_item(item), mp_front(p_front), mp_next(p_next)    {}        T m_item;    dc_list_node<T> *mp_front;    dc_list_node<T> *mp_next;private:    dc_list_node(const dc_list_node<T> &);    dc_list_node<T> &operator=(const dc_list_node<T> &);};

void dc_list_erase_node_with_equal_value(dc_list_node<int> *&p_headA, dc_list_node<int> *&p_headB){    if (p_headA == NULL || p_headB == NULL)        return;    // A_vec存储链表A的元素值,B_vec存储链表B的元素值    std::vector<int> A_vec, B_vec, common_vec;    dc_list_node<int> *p_node;    A_vec.push_back(p_headA->m_item);    p_node = p_headA->mp_next;    while (p_node != p_headA)    {        A_vec.push_back(p_node->m_item);        p_node = p_node->mp_next;    }    B_vec.push_back(p_headB->m_item);    p_node = p_headB->mp_next;    while (p_node != p_headB)    {        B_vec.push_back(p_node->m_item);        p_node = p_node->mp_next;    }    // 将A_vec和B_vec排序    std::sort(A_vec.begin(), A_vec.end());    std::sort(B_vec.begin(), B_vec.end());    // 合并A_vec和B_vec相同的元素值到common_vec    std::size_t i = 0, j = 0;    while (i < A_vec.size() && j < B_vec.size())    {        if (A_vec[i] < B_vec[j])            ++i;        else if (A_vec[i] > B_vec[j])            ++j;        else        {            common_vec.push_back(A_vec[i]);            ++i;            ++j;        }    }    // 分别遍历链表A和B,删除在元素值common_vec中的结点    p_node = p_headA;    for (i = 0; i < A_vec.size(); ++i)    {        if (std::binary_search(common_vec.begin(), common_vec.end(), p_node->m_item)) // 当前结点是值相同的结点        {            if (p_node == p_headA && p_node->mp_next == p_headA) // 链表中只有头结点一个结点            {                delete p_headA;                p_headA = NULL;                break;            }            else if (p_node == p_headA) // 当前结点是头结点            {                p_headA = p_node->mp_next;                p_headA->mp_front = p_node->mp_front;                p_node->mp_front->mp_next = p_headA;                p_node = p_headA;            }            else            {                dc_list_node<int> *p_next = p_node->mp_next, *p_front = p_node->mp_front;                p_next->mp_front = p_front;                p_front->mp_next = p_next;                delete p_node;                p_node = p_next;            }        }        else            p_node = p_node->mp_next;    }    p_node = p_headB;    for (i = 0; i < B_vec.size(); ++i)    {        if (std::binary_search(common_vec.begin(), common_vec.end(), p_node->m_item)) // 当前结点是值相同的结点        {            if (p_node == p_headB && p_node->mp_next == p_headB) // 链表中只有头结点一个结点            {                delete p_headB;                p_headB = NULL;                break;            }            else if (p_node == p_headB) // 当前结点是头结点            {                p_headB = p_node->mp_next;                p_headB->mp_front = p_node->mp_front;                p_node->mp_front->mp_next = p_headB;                p_node = p_headB;            }            else            {                dc_list_node<int> *p_next = p_node->mp_next, *p_front = p_node->mp_front;                p_next->mp_front = p_front;                p_front->mp_next = p_next;                delete p_node;                p_node = p_next;            }        }        else            p_node = p_node->mp_next;    }}