12.3 其它链表技巧
160. Intersection of Two Linked Lists
题目描述
给定两个链表,判断它们是否相交于一点,并求这个相交节点。
输入输出样例
输入是两条链表,输出是一个节点。如无相交节点,则返回一个空节点。
Input:
A: a1 -> a2
|
v
c1 -> c2 -> c3
^
|
B: b1 -> b2 -> b3
Output: c1
题解
假设链表 A 的头节点到相交点的距离是 a,链表 B 的头节点到相交点的距离是 b,相交点到链表终点的距离为 c。我们使用两个指针,分别指向两个链表的头节点,并以相同的速度前进,若到达链表结尾,则移动到另一条链表的头节点继续前进。按照这种前进方法,两个指针会在 a + b + c 次前进后同时到达相交节点。
- C++
- Python
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode *l1 = headA, *l2 = headB;
while (l1 != l2) {
l1 = l1 != nullptr ? l1->next : headB;
l2 = l2 != nullptr ? l2->next : headA;
}
return l1;
}
def getIntersectionNode(
headA: ListNode, headB: ListNode
) -> Optional[ListNode]:
l1 = headA
l2 = headB
while l1 != l2:
l1 = l1.next if l1 is not None else headB
l2 = l2.next if l2 is not None else headA
return l1