面试题52. 两个链表的第一个公共节点

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

输入：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 个节点。

思路1：双栈法

经过分析发现，如果两个链表有公共节点，那么公共节点出现在两个链表的尾部。如果从两个链表的尾部开始向前比较，那么最后一个相同的节点就是要找的节点。这种思路可使用栈实现，分别把两个链表的节点放在两个栈里，这样两个栈表的尾节点就位与两个栈的栈顶，接下来比较两个栈顶的节点是否相同，如果相同，弹出，比较下一对，直到找到最后一对相同的节点。
这种思路需要两个辅助栈，若原链表的长度为m，n，则时间复杂度O(m+n)，空间复杂度O(m+n)

class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        if not headA or not headB:
            return None
        stackA = []
        stackB = []
        while headA:
            stackA.append(headA)
            headA = headA.next
        while headB:
            stackB.append(headB)
            headB = headB.next
        
        res = None
        while stackA[-1]==stackB[-1]:
            res = stackA[-1]
            stackA.pop()
            stackB.pop()
            if not stackA or not stackB:
                break
        return res

思路2：双指针法

首先遍历两个链表得到它们的长度，这样就能知道哪个链表比较长，以及长的链表比短的链表多几个节点。在第二次遍历时，在较长的链表上先走多的那几步，接着同时在两个链表上遍历，找到的第一个相同的节点就是它们的第一个公共节点。

思路2变种

使用两个指针 node1，node2 分别指向两个链表 headA，headB 的头结点，然后同时分别逐结点遍历，当 node1 到达链表 headA 的末尾时，重新定位到链表 headB 的头结点；当 node2 到达链表 headB 的末尾时，重新定位到链表 headA 的头结点。
这样，当它们相遇时，所指向的结点就是第一个公共结点。

# 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:
        node1, node2 = headA, headB
        
        while node1 != node2:
            node1 = node1.next if node1 else headB
            node2 = node2.next if node2 else headA

        return node1