在两个有序链表中查找第K大元素。

这是一次面试的时候，别人问我的，当时回答的不太好。



题目描述：



即在两个有序链表中查找第k大的元素，相等的元素做一个元素处理。



{1，3，5}， {2，4，6}，这是简单的情况，查找第2大的元素则是 2。



{1，3， 3，3， 5}，{2，3，3，4，6}，稍微复杂，查找第3大元素则是3，第4大元素 则是4.



解答：



其实我们可以借鉴合并有序链表的方法来解决这个问题。





算法实现如下：

struct node{  
    int data;  
    node* next;  
};  
typedef node* List;  
int findKthElement(List &list1, List &list2, int k)  
{  
    List pList1 = list1;  
    List pList2 = list2;  
    List pCurNode = NULL;
    int count = 0;
    while(pList1 && pList2)  
    {  
        if (pList1->data < pList2->data)  
        {    
            pCurNode = pList1;  
            pList1 = pList1->next;  
            count++;
        }  
        else if (pList1->data > pList2->data)
        {  
            pCurNode = pList2;  
            pList2 = pList2->next;  
            count++;
        } 
        else
        {
            count++;
            pCurNode = pList1;
            int temp = pList1->data;
            List p = pList1->next;
            while(p)
            {
                if (p->data != temp)
                {
                    pList1 = p;
                    break;
                }
                p = p->next;
            }
            p = pList2->next;
            while(p)
            {
                if (p->data != temp)
                {   
                    pList2 = p;
                }
                p = p->next;
            }
        }
  
        if (count == k)
        {
            return pCurNode->data;
        }
    }  
    if (pList1 == NULL && pList2 == NULL)
    {
        return -1;
    }
    List rest = (pList1 != NULL) ? pList1 : pList2;
    while(rest)
    {
        count++;
        if (count == k)
        {
            return rest->data;
        }
        rest = rest->next;
    }
    return  -1;
}