编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 c1 开始相交。
示例 1:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Reference of the node with value = 8
输入解释:相交节点的值为 8 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
示例 2:
输入:intersectVal = 2, listA = [0,9,1,2,4], listB = [3,2,4], skipA = 3, skipB = 1
输出:Reference of the node with value = 2
输入解释:相交节点的值为 2 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [0,9,1,2,4],链表 B 为 [3,2,4]。在 A 中,相交节点前有 3 个节点;在 B 中,相交节点前有 1 个节点。
示例 3:
输入:intersectVal = 0, listA = [2,6,4], listB = [1,5], skipA = 3, skipB = 2
输出:null
输入解释:从各自的表头开始算起,链表 A 为 [2,6,4],链表 B 为 [1,5]。由于这两个链表不相交,所以 intersectVal 必须为 0,而 skipA 和 skipB 可以是任意值。
解释:这两个链表不相交,因此返回 null。
注意:
- 如果两个链表没有交点,返回 null.
- 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
思路
1.暴力匹配
最简单、最容易的方法,但也是最差的解法。效率非常低。
内外嵌套循环依次查找是否两个链表是否会指向同样的节点。
时间复杂度:O(mn)
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if(headA == null || headB == null){
return null;
}
ListNode a = headA;
ListNode b = headB;
while(a != null){
while(b != null){
if(b == a){
return b;
}else{
b = b.next;
}
}
a = a.next;
b = headB;
}
return null;
}
2.哈希表
先遍历一个链表,将其所有元素存入哈希表,
然后依次遍历第二个链表,看节点在哈希表中是否已经存在。
相比暴力匹配可可将时间复杂度降到O(m+n),但是消耗额外空间。
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if(headA == null || headB == null){
return null;
}
ListNode a = headA;
ListNode b = headB;
Map<ListNode,Integer> map = new HashMap<>();
while(a != null){ //依次存入第一个链表的元素
map.put(a,1);
a = a.next;
}
while(b != null){
if(map.containsKey(b)){ //包含b,说明b这个节点在a中也存在,就是相交的节点
return b;
}
b = b.next;
}
return null;
}
3.截取长度
假象一下:如果两个链表长度是一样的,那么会很简单,我们从头到尾一个一个依次对比就行了。
但是链表长度不一样,就想办法把他转化成一样的长度。
如果他们指向了同一个节点,那么后面长度一定是一样的,那么先办法把前面两个对齐。
做法:
- 先遍历两个链表长度,length1,length2,
- 假如length1>length2,那么第一个链表从length1-length2处开始,第二个链表从头开始依次比较是否指向同一个元素。
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if(headA == null || headB == null){
return null;
}
ListNode a = headA;
ListNode b = headB;
int lengthA = 0; //记录a链表长度
int lengthB = 0; //记录b链表长度
while (a != null){
lengthA++;
a = a.next;
}
while(b != null){
lengthB++;
b = b.next;
}
if(lengthA>lengthB){
a = headA;
for(int i = 0;i<lengthA-lengthB;i++){
a = a.next;
}
b = headB;
while(a != null){
if(a == b){
return a;
}
a = a.next;
b = b.next;
}
}else{
b = headB;
for (int i = 0;i<lengthB-lengthA;i++){
b = b.next;
}
a = headA;
while(b != null){
if(b == a){
return b;
}
b = b.next;
a = a.next;
}
}
return null;
}
4.双指针法(不太容易想到)
a、b是两个链表的头,依次遍历,
a遍历结束就去从b链表头结点开始。
b遍历结束就去a链表头结点开始。
当两个指针相等就是出现交叉。
这种方法可以这么理解。两个链表长度是不均等的,但是遍历一遍再去遍历另外一个,这样链表长度差就消失了,抽象的可以理解成两个一模一样长的链表,从头结点一个一个依次比较,看是否存在相同节点。
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if (headA == null || headB == null) {
return null;
}
ListNode a = headA;
ListNode b = headB;
while(a != b){
if(a != null){
a = a.next;
}else{
a = headB;
}
if(b != null){
b = b.next;
}else{
b = headA;
}
}
return a; //如果出现交叉节点,最后在交叉节点相遇。如果没有,a==b==null,最终返回null
}