1. 程式人生 > >雅虎面試題─有兩個雙向迴圈連結串列A,B,知道其頭指標為:pHeadA,pHeadB,請寫一函式將兩連結串列中data值相同的結點刪除

雅虎面試題─有兩個雙向迴圈連結串列A,B,知道其頭指標為:pHeadA,pHeadB,請寫一函式將兩連結串列中data值相同的結點刪除

有雙向迴圈連結串列結點定義為:

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;
    }
}