题目描述:
编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 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) 内存。
解题思路:
其实我想的很简单,因为两个链表不一样长,那么第一次遍历,分别记录两个链表的长度。然后将长度作差,让更长的链表先走相差的那些步,然后两个链表一起走,然后相遇的第一个结点即为两个链表相交的交点。
代码实现:
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if(headA == null || headB == null) {
return null;
}
ListNode A = headA;
ListNode B = headB;
int lengthA = 0;
int lengthB = 0;
while(A != null && B!= null) {
A = A.next;
B = B.next;
lengthA++;
lengthB++;
}
while(A != null) {
A = A.next;
lengthA++;
}
while(B != null) {
B = B.next;
lengthB++;
}
int step = lengthA - lengthB;
if (step > 0) { //说明A比B长
A = headA;
B= headB;
}else {
A = headB; //B比A长
B= headA;
step *= -1; //将step变为正值
}
while(step > 0) {
A = A.next;
step--;
}
while(A != B) {
A = A.next;
B = B.next;
}
return A;
}
自己又修改了一下,不用记录两个链表的长度,只需要两个链表一起遍历,其中一个走到末尾,另一个还继续走,这时再开始记录长度,也就是说只记录链表余出来的长度,并且记录是哪条链表更长,之后同上。
附上代码:
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if(headA == null || headB == null) {
return null;
}
ListNode A = headA;
ListNode B = headB;
int length = 0; //末尾余出来的长度
Boolean flag = false; //默认A链表更长
while(A != null && B!= null) {
A = A.next;
B = B.next;
}
while(A != null) {
A = A.next;
length++;
}
while(B != null) {
flag = true; //B链表更长
B = B.next;
length++;
}
A = headA;
B = headB;
if(length > 0 && flag) { //B链表先走length步
while(length > 0) {
B = B.next;
length--;
}
}else if (length > 0){ //A链表先走length步。
while(length > 0) {
A = A.next;
length --;
}
}
//两链表同走
while(A != B) {
A = A.next;
B = B.next;
}
return A;
}