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

题目描述

输入两个链表，找出它们的第一个公共结点。

解题思路

首先，如果都是空链表，那么肯定返回空。两个单链表假设有公共节点，那么肯定是“Y”字型的。

• 先计算两个链表的长度，求出长度差
• 根据长度差，后移长链表的头指针，使得两个链表的现在的头指针的位置距离为节点一样
• 同步移动，直到指针相等，是Y型的连接处，也就是公共点；注意没有公共点的情况。

AC代码

class Solution {
  public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        if(pHead1 == nullptr || pHead2 == nullptr) {  // 日常判空
            return nullptr;
        }
        int len1 = 0, len2 = 0;
        // 计算链1的长度
        ListNode* p = pHead1;
        while(p != nullptr) {
            ++len1;
            p = p->next;
        }
        // 计算链2的长度
        p = pHead2;
        while(p != nullptr) {
            ++len2;
            p = p->next;
        }
        // p1是长链，p2是短链
        auto p1 = (len1 > len2) ? pHead1 : pHead2;
        auto p2 = (len1 <= len2) ? pHead1 : pHead2;
        int len = max(len1, len2) - min(len1, len2);
        while(len > 0) {  // 长链后移，同步长度
            p1 = p1->next;
            --len;
        }
        // 寻找公共节点，注意没有的情况
        while(p1 != p2 && p1 != nullptr) {
            p1 = p1->next;
            p2 = p2->next;
        }
        return p1;
    }
};