剑指offer52.两个链表的第一个公共节点
输入两个链表,找出它们的第一个公共节点。
如下面的两个链表:
示例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 个节点。
解题思路
采用双指针方法来求解这道题,求出两个链表的公共节点其实就是要求这两个链表的长度差是多少,找到了两条链表的长度差d后,让较长的那条链表先走d的距离,这样两条链表就相当于在同一起跑线,接下来对这两条链表同时进行遍历,便可以找到两条链表的公共节点。
代码
public class Solution {
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
ListNode l1 = headA;
ListNode l2 = headB;
while (l1!=null && l2!=null){
l1=l1.next;
l2=l2.next;
}
int cnt=0;
if(l1==null){
while (l2!=null){
l2=l2.next;
cnt++;
}
while(cnt!=0){
headB=headB.next;
cnt--;
}
while (headA!=headB){
headA=headA.next;
headB=headB.next;
}
return headA;
}else if(l2==null){
while (l1!=null){
l1=l1.next;
cnt++;
}
while (cnt!=0){
headA=headA.next;
cnt--;
}
while (headA!=headB){
headA=headA.next;
headB=headB.next;
}
return headB;
}else {
return headA;
}
}
}