[剑指offer]两个链表的第一个公共结点

两个链表的第一个公共结点

题目

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

当不存在公共节点时，返回空节点。

样例

给出两个链表如下所示：

A： a1 → a2
                      ↘
                           c1 → c2 → c3
                         ↗
B: b1 → b2 → b3

输出第一个公共节点c1

题解1

利用unordered_map记录 A 链表，然后，遍历 B 链表，返回 B 链表中第一个在unordered_map中出现的结点即为答案

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *findFirstCommonNode(ListNode *headA, ListNode *headB) {
        auto t = headA;
        unordered_map<ListNode*, int> umap;
        
        while (t) umap[t] ++, t = t->next;
        
        t = headB;
        while (t) {
            if (umap.count(t)) return t;
            t = t->next;
        }
        return NULL;
    }
};

题解2

两个链表中不相同的部分分别为a和b，公共部分为c，由此可以得到结论：a + c + b = b + c + a
设置两个指针一起走，a走到尾就转向b， b走到尾转向a，则在公共部分相遇，即为答案

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *findFirstCommonNode(ListNode *headA, ListNode *headB) {
        auto p = headA, q = headB;
        while (p != q) {
            if (p) p = p->next;
            else p = headB;
            if (q) q = q->next;
            else q = headA;
        }
        return p;
    }
};