尋找連結串列相交結點問題
作者:Grey
原文地址:
題目描述
給定兩個可能有環也可能無環的單連結串列,頭節點head1和head2。請實現一個函式,如果兩個連結串列相交,請返回相交的 第一個節點。如果不相交,返回 null。
要求:如果兩個連結串列長度之和為N,時間複雜度請達到O(N)
,額外空間複雜度請達到O(1)
。
類似問題
本題主要的難點是要分析所有可能的情況,因為題目中提到「可能有環也可能無環」。
主要思路
先看大的情況,有如下三種情況
第一種情況:兩個連結串列均無環;
第二種情況:兩個連結串列均有環;
第三種情況:一個有環,一個無環。
首先,第三種情況下,兩個連結串列一定不相交。針對第一種情況,就是尋找連結串列的入環節點和相交節點問題中提到LeetCode 160. Intersection of Two Linked Lists,現在只分析第二種情況。
由於兩個連結串列都有環,兩個連結串列如果相交,一定只有如下三種情況
情況1:兩個連結串列獨立不相交
情況2:兩個連結串列的入環結點是同一個
情況3:兩個連結串列的入環結點不是同一個,此時任意一個連結串列的入環結點都是相交結點。
先從最簡單的情況1和情況3進行分析,情況一發生的條件是:兩個連結串列的入環結點(loop1,loop2)不是同一個,判斷條件很簡單,就是從任意一個連結串列的入環結點開始遍歷一圈,如果都沒有遇到另外一個連結串列的入環結點, 兩個連結串列不相交,屬於情況1;
如果從任意一個連結串列的入環結點開始遍歷一圈,遇到了另外一個連結串列的入環結點,則說明兩個連結串列相交,屬於情況3,且任意一個連結串列的入環結點都是相交結點。
最後分析情況2,兩個連結串列的入環結點如果是同一個,可以記錄兩個連結串列的差值,然後讓短連結串列先走差值步以後,長短連結串列同時開始走,相遇的結點就是第一個相交結點。
完整程式碼見:
public class Code_FindFirstIntersectNode {
public static class List {
public int val;
public List next;
public List(int v) {
val = v;
}
}
public static List getIntersectNode(List head1, List head2) {
if (head1 == null || head2 == null) {
return null;
}
// 兩個均無環
List loop1 = getLoopNode(head1);
List loop2 = getLoopNode(head2);
if (loop1 == null && loop2 == null) {
return noLoop(head1, head2);
}
// 兩個均有環
if (loop1 != null && loop2 != null) {
return bothLoop(head1, loop1, head2, loop2);
}
// 一個有環一個無環 ,不可能相交
return null;
}
// 找到連結串列第一個入環節點,如果無環,返回null
public static List getLoopNode(List head) {
if (head == null || head.next == null || head.next.next == null) {
return null;
}
// 慢指標 在第一個節點位置
List slow = head.next;
// 快指標,在第二個節點的位置
List fast = head.next.next;
while (slow != fast) {
if (fast.next == null || fast.next.next == null) {
return null;
}
// 快指標每次走兩步
fast = fast.next.next;
// 慢指標每次走一步
slow = slow.next;
}
// 兩個指標遇上了,說明有環
// 讓快指標回到頭部, 慢指標停在原地
fast = head;
while (fast != slow) {
fast = fast.next;
slow = slow.next;
}
// 快指標每次走一步,慢指標每次走一步,遇上後,就是入環節點處
return slow;
}
// 如果兩個連結串列都無環,返回第一個相交節點,如果不想交,返回null
public static List noLoop(List head1, List head2) {
if (head1 == null || head1 == null) {
return null;
}
// 判斷兩個連結串列的長度
int n = 0;
List t1 = head1;
List t2 = head2;
while (t1.next != null) {
n++;
t1 = t1.next;
}
while (t2.next != null) {
n--;
t2 = t2.next;
}
// 兩個連結串列的末節點不相等
if (t2 != t1) {
return null;
}
// 記錄長的連結串列頭節點
List longer = n > 0 ? head1 : head2;
// 記錄短的連結串列頭節點
List shorter = longer == head1 ? head2 : head1;
// 先讓長連結串列走一段距離(這段的長度就是長連結串列和短連結串列的長度差)
int gap = Math.abs(n);
while (gap != 0) {
gap--;
longer = longer.next;
}
// 然後長連結串列和短連結串列同時開始走,直到相等的節點即為交點
while (longer != shorter) {
longer = longer.next;
shorter = shorter.next;
}
return shorter;
}
// 兩個有環連結串列,返回第一個相交節點,如果不想交返回null
public static List bothLoop(List head1, List loop1, List head2, List loop2) {
// 只有兩種情況
if (loop1 == loop2) {
// 1. 未入環就相交
// 這種情況下,兩個連結串列的入環節點是一樣
int n = 0;
List t1 = head1;
List t2 = head2;
while (t1 != loop1) {
n++;
t1 = t1.next;
}
while (t2 != loop2) {
n--;
t2 = t2.next;
}
List longer = n > 0 ? head1 : head2;
List shorter = longer == head1 ? head2 : head1;
n = Math.abs(n);
while (n != 0) {
n--;
longer = longer.next;
}
while (longer != shorter) {
longer = longer.next;
shorter = shorter.next;
}
return shorter;
} else {
// 2. 共用環,不在入環處相交,隨便一個連結串列的入環點就是交點
// loop1 != loop2
// 從loop1開始,轉一圈回到loop1
// 如果都沒有遇到loop2,則不相交
// 如果遇到了loop1,則交點為loop1或者loop2都可以
List t1 = loop1.next;
while (t1 != loop1) {
if (t1 == loop2) {
return loop1;
}
t1 = t1.next;
}
return null;
}
}
}