一、题目
输入两个链表,找出它们的第一个公共结点。
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
A, B = headA, headB
while A != B:
A = A.next if A else headB
B = B.next if B else headA
return A
二、解析
我们假设有两个链表 `A` 和 `B`,并且它们在某个节点相遇。以下是一个示意图:
A: 1 -> 2 -> 3 \
\
6 -> 7 -> 8
/
B: 4 -> 5 /
在这个示意图中,链表 `A` 和链表 `B` 在节点 `6` 处相遇。现在,让我们用代码中的指针 `A` 和 `B` 来模拟这个相遇的过程。
初始状态:
A: 1 -> 2 -> 3 -> 6 -> 7 -> 8
B: 4 -> 5 -> 6 -> 7 -> 8
代码中的 while 循环开始执行:
1. 第一次循环:
A = A.next # 移动 A 指针到下一个节点(节点 2)
B = B.next # 移动 B 指针到下一个节点(节点 5)
A: 2 -> 3 -> 6 -> 7 -> 8
B: 5 -> 6 -> 7 -> 8
2. 第二次循环:
A = A.next # 移动 A 指针到下一个节点(节点 3)
B = B.next # 移动 B 指针到下一个节点(节点 6)
A: 3 -> 6 -> 7 -> 8
B: 6 -> 7 -> 8
3. 第三次循环:
A = A.next # 移动 A 指针到下一个节点(节点 6)
B = B.next # 移动 B 指针到下一个节点(节点 7)
A: 6 -> 7 -> 8
B: 7 -> 8
A为空后去B循环,B为空后去A循环:
A + B: 1 -> 2 -> 3 -> 6 -> 7 -> 8 -> 4 -> 5 -> 6 -> 7 -> 8
B + A: 4 -> 5 -> 6 -> 7 -> 8 -> 1 -> 2 -> 3 -> 6 -> 7 -> 8
第九次循环时:
此时,`A` 和 `B` 指针同时指向节点 `6`,循环结束。根据代码,最终返回 `A` 或 `B` 中的任意一个,都是指向相交的节点 `6`。在这个例子中,返回 `A` 或 `B` 都是一样的,因为它们在相交节点上相等。