两个链表的第一个公共结点
题目
输入两个链表,找出它们的第一个公共结点。
当不存在公共节点时,返回空节点。
样例
给出两个链表如下所示:
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;
}
};