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

Posted 2021-06-21 风去幽墨

题目链接：

https://leetcode-cn.com/problems/liang-ge-lian-biao-de-di-yi-ge-gong-gong-jie-dian-lcof/

题意：

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

题解：

经典面试题。
解法：
链表顺着走一遍，走法是当一个链表走到尾节点时，跳转到另一个链表的头节点。
正确性：
记两个链表的不重复部分的长度为lena和lenb，公共部分的长度为len。
那么按照上述走法，两者走的长度分别为lena+len+lenb和lenb+len+lena，所以最终会走到同一个点上去，而这个点就是两条链表的第一个公共节点。
注意：
两条链表没有公共节点的情况。可以看代码部分的正解与错解的区别在于链表到达尾节点时的判断条件（pA == None | pA.next == None ）。
这样写的原因是：假设两个链表没有公共节点的话，第一种写法相当于是在每个链表的结尾添加了一个空节点，这样即使没有公共节点，在走完lena+lenb后依据会同时指向空节点，就可以跳出循环，且返回空；而第二种写法中，则是在没有公共节点的情况下，两个指针在第一次跳转后，不断在另一个链表上循环跑（例如 pA会一直在链表B上循环跑），导致不断循环，无法得正解。

代码：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
#正解
class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        pA,pB= headA,headB
        while pA != pB:
            pA = headB if not pA else pA.next
            pB = headA if not pB else pB.next
        return pA
#错解
class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        pA,pB= headA,headB
        while pA != pB:
            pA = headB if not pA.next else pA.next
            pB = headA if not pB.next else pB.next
        return pA