原题链接:https://leetcode-cn.com/problems/liang-ge-lian-biao-de-di-yi-ge-gong-gong-jie-dian-lcof/
题目
输入两个链表,找出它们的第一个公共节点。
如下面的两个链表:
在节点 c1 开始相交。
示例 1:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Reference of the node with value = 8
输入解释:相交节点的值为 8 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
思路
先把链表A都加到哈希表里面,然后遍历B找到第一个相同的节点。
或者A和B都先移动到最后一个节点,若不想等则表示没有共同节点,若相等则回退到第一个相等的节点。这样子倒也不复杂,空间占用也不多。
哈希表的话比较好写,就写一下第二个思路的。
half an hour later,事后证明不好写,递归也不好写,看下题解。
很巧妙的一个做法,核心思想就是让两个链表都走相同的路径。两个指针a和b分别从链表A和链表B的头节点出发,当a走到链表A的尾部的时候,从链表B的头节点再开始走,指针b同理。也就是说若两次都走完的话a和b都走了(A+B)的长度的路程,那他们如果相遇,第一次相遇的地方就是两个链表的第一个公共节点。
题解的代码太优美了,自己虽然按照思路实现了但是判断太多了。但事实上若该两个链表不相交,那么指针第二次遍历链表的时候最后会同时为空,那么就会返回空,实在是太牛了。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode* ptrA = headA;
ListNode* ptrB = headB;
while(ptrA != ptrB) {
ptrA = ptrA != nullptr ? ptrA->next : headB;
ptrB = ptrB != nullptr ? ptrB->next : headA;
}
return ptrA;
}
};