LeetCode解题心得——相交链表（python）

题目

地址：https://leetcode-cn.com/problems/intersection-of-two-linked-lists/
编写一个程序，找到两个单链表相交的起始节点。

要求：

• 如果两个链表没有交点，返回 null.
• 在返回结果后，两个链表仍须保持原有的结构。
• 可假定整个链表结构中没有循环。
• 程序尽量满足 O(n) 时间复杂度，且仅用 O(1) 内存

思路

设定两个指针，p1，p2，分别从A和B的表头开始跑，当p1到A的尾部时，从B的头部开始跑。同样p2到B的尾部时，从A的头部开始跑。两个指针会在交点处会和。如果A和B在交点之前的节点数一样，那么两个指针会在首次遍历时会和，如果不一样，两指针会在第二次遍历时会和。
如果A、B没有交点，p1，p2会同时为None，因为都跑了A+B的距离。

class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        p1 = headA
        p2 = headB
        while(p1 != p2):
            p1 = p1.next if p1 != None else headB
            p2 = p2.next if p2 != None else headA
        return p1