题目
编写一个程序,找到两个单链表相交的起始节点。
注意:
- 如果两个链表没有交点,返回 null.
- 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
解题思路
由于题目的限制,使用暴力法的话,需要对A链表的每个节点都去B链表中找有没有相同的节点。情况最坏的时候,可能两个链表在最后一个节点相交。这样复杂度就有不符合要求了
这个时候需要逆向思维,咱们从链表的头部去搜寻,那么从链表的尾部可以吗?
通过观察,发现其实链表在相交的那一刻起,后面的长度就是一模一样的了
所以思路如下:
- 循环遍历找到两个链表的长度
- 将长一点的那个链表提前遍历到和短一点一样长
- 这样问题就转化成了,两个等长的链表同步地往后面运动
代码
public class Solution {
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
ListNode runnerA = headA, runnerB = headB;
int lenA = 0, lenB = 0;
// 找到链表的长度
while(runnerA != null){
lenA++;
runnerA = runnerA.next;
}
while(runnerB != null){
lenB++;
runnerB = runnerB.next;
}
// 让 headA指针指向长的链表
if(lenB > lenA){
ListNode temp = headA;
headA = headB;
headB = temp;
}
// 让长的链表先跑一段距离
int len = Math.abs(lenB-lenA);
while(len-->0){
headA = headA.next;
}
// 同时跑
while(headA != headB){
headA = headA.next;
headB = headB.next;
}
return headA;
}
}
代码的可读性还好,就是有点长
浪漫的解题方法
如图
解读:
就是 a和 b走到尽头的时候,如果没有碰到彼此
那么 a、b就会去走对方走过的路,此时a、b就会碰到对方了
public class Solution {
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
ListNode ha = headA, hb = headB;
while (ha != hb) {
// 这里的 null很关键,
// 如果两个链表没有交点的时候,这里的null就起了大作用
if(ha != null)
ha = ha.next;
else
ha = headB;
if(hb != null)
hb = hb.next;
else
hb = headA;
}
return ha;
}
}
这里需要注意细节,如何两个链表没有交点的话,当两者都为 null的时候,跳出循环