题目
编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
思路一:用哈希表来做,先遍历链表A,再把各个节点信息存到哈希表中。再遍历链表B,如果第一个加不进去的节点就是相交节点。如果都没有相交,即都存入成功,我们就返回null
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
Set<ListNode> set = new HashSet<>();
// 先遍历链表A
while(headA!=null){
set.add(headA);
headA = headA.next;
}
// 再遍历链表B
while(headB!=null){
if(!set.add(headB)){
return headB;
}
headB = headB.next;
}
return null;
}
思路二:A走完自己的路去走B的路,B走完自己的路去走A的路,第一个相遇的节点,即A和B相等的时候,我们就认为找到了。即使没有相遇,由于A+B的长度是一样的,也就是他俩最后一定会走到null,这时候,返回null即可。
public ListNode getIntersectionNode1(ListNode headA, ListNode headB){
ListNode temp1 = headA;
ListNode temp2 = headB;
while(temp1!=temp2){
if(temp1!=null){
temp1 = temp1.next;
}else {
temp1 = headB;
}
if(temp2!=null){
temp2 = temp2.next;
}else {
temp2 = headA;
}
}
return temp1;
}