剑指Offer两个链表的第一个公共结点

题目描述
输入两个无环的单链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)

解法
首先考虑特殊情况,两链表有一个为空,返回空指针
分别求两个链表的长度,那么有可能的第一个公共结点是:长链表走到从长链表结点数-短链表结点数的那个结点,然后短链表从头开始,两链表依次走一个结点,比较是否相同

class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        if(pHead1 == pHead2) return pHead1;
        if(pHead1 == NULL || pHead2 == NULL) return nullptr;
        
        ListNode* p1, *p2;
        p1 = pHead1;
        p2 = pHead2;
        int i = 0, j = 0;
        //获取两个链表的长度
        while(p1 != NULL)
        {
            i++;
            p1 = p1 ->next;
        }
        
        while(p2 != NULL)
        {
            j++;
            p2 = p2 ->next;
        }
        
        int k;
        if(i >= j) k = i -j;
        else k = j - i;
        
        p1 = pHead1;
        p2 = pHead2;
        //让更长的链表走到第一个可能公共结点处
        if(i >= j)
        {
            i = 0;
            while(i < k)
            {
                i++;
                p1 = p1->next;
            }
            while(p1 != p2 && p1 != NULL && p2 != NULL)
            {
                p1 = p1->next;
                p2 = p2->next;
            }
            return p1;
        }
        else
        {
            j = 0;
            while(j < k)
            {
                j++;
                p2 = p2->next;
            }
            while(p1 != p2 && p1 != NULL && p2 != NULL)
            {
                p1 = p1->next;
                p2 = p2->next;
            }
            return p1;
        }
    }
};

时间复杂度O(n)
空间复杂度O(1)


版权声明:本文为wjx19960817原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。