编写一个程序,找到两个单链表相交的起始节点。
例如,下面的两个链表:
A: a1 → a2
↘
c1 → c2 → c3
↗
B: b1 → b2 → b3
在节点 c1 开始相交。
注意:
- 如果两个链表没有交点,返回
null
. - 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
思路:如果没有 O(n) 时间复杂度,且仅用 O(1) 内存的条件,这个题很容器写,但是因为对于时间复杂度和空间复杂度有要求,所以,我们应该再做思考,我们知道,如果两个链表相交,从相交点开始,以后的元素都是相同的,我们设A链表中,不重复的为listA,设A链表中,不重复的为listB,相同的那部分为AequalB。
那么就有listA+AequalB+listB=listB+AequalB+listA,当然这也是显然易见的,所以我们就可以找到交点了,
例如,对于A链表,我们a1 → a2 → c1→c2 → c3→b1 → b2 → b3→c1
对于B链表,我们b1 → b2 → b3→c1 → c2 → c3→a1 → a2→c1
可以发现,我们从A链表开始,当A链表遍历完后,再从B链表头节点找, 对于B,我们从B链表开始,当B链表遍历完后,再从A链表头节点找,这样会在相交的地方,两值相等。
public class Main {
public static void main(String[] args) {
ListNode node = new ListNode(1);
node.next=new ListNode(2);
ListNode n1=new ListNode(4);//A链表
n1.next=new ListNode(3);
n1.next.next=node;
ListNode n2=new ListNode(7);//B链表
n2.next=new ListNode(6);
n2.next.next=new ListNode(5);
n2.next.next.next=node;
ListNode node1 = getIntersectionNode(n1, n2);
System.out.println(node1==null?"null":node1.val);
}
public static ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if(headA==null || headB==null){
return null;
}
ListNode listA=headA;
ListNode listB=headB;
while(listA!=listB){
listA= (listA==null) ? headB : listA.next; //当A链表遍历完,开始遍历B链表
listB= (listB==null) ? headA : listB.next; //当B链表遍历完,开始遍历A链表
}
return listA;
}
static class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
next = null;
}
}
}