想要精通算法和SQL的成长之路 - 相交链表
前言
一. 相交链表(双指针)
原题链接
思路如下:
1.我们假设 headA
链表的长度为 a
。headB
链表的长度为b
。两个链表的公共部分长度为c
(如果存在),公共节点为node
。
- 头结点
headA
到node
前,有a-c
个节点。 - 头结点
headB
到node
前,有b-c
个节点。
2.那么我们用两个指针,A
和B
。分别指向两个链表的头节点。
- 指针
A
走完链表headA
再走链表headB
。走到node
节点的时候,走过的步数为:a + (b - c)
- 指针
B
走完链表headB
再走链表headA
。走到node
节点的时候,走过的步数为:b + (a - c)
首先,两个指针走过的步数一定是相等的,但是对于返回结果而言,我们可以根据c
来区分:
- 如果两个链表存在相交链表,那么
c
肯定非空,此时指针A
或者B
都是指向node
节点。 - 如果两个链表无相交部分,那么
c
的值为0,也就是说此时A
或者B
的指向是null
节点。
因此我们选哪个都可以,最后返回任意一个指针即可,代码如下:
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
ListNode a = headA, b = headB;
while (a != b) {
// 如果指针a为null,那么继续遍历headB链表,否则继续遍历headA链表。指针b同理
a = a == null ? headB : a.next;
b = b == null ? headA : b.next;
}
return a;
}