Java算法 相交链表--找出并返回两个单链表相交的起始节点

小王的Java刷题日记Day2

记录刷题过程，作为笔记和分享，坚持每天刷题，每天进步，编程语言为Java。

题目：给你两个单链表的头节点 headA 和 headB ，请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点，返回 null 。

思路一：双指针法       时间复杂度O(m+n)

链表A和链表B如果有相交节点时，若A、B相交之后的节点长度为x,链表A单独节点长度为a，链表B单独的节点长度为b，那么从头节点同时出发，遍历完链表A时走过的长度为a+x，则让PA指针指向链表B的头节点继续移动。遍历完链表B时走过的长度为b+x,则让PB指针指向链表A的头节点继续移动。若有相交节点，则当PA继续走完链表B单独的节点长度时，PB此时也走完了链表A单独的节点长度，即a+x+b=b+x+a.。

1、每步操作需要同时更新指针 pA 和 pB。

2、如果指针 pA不为空，则将指针 pA 移到下一个节点；如果指针 pB 不为空，则将指针 pB 移到下一个节点。

3、如果指针 pA为空，则将指针 pA移到链表 headB 的头节点；如果指针 pB 为空，则将指针 pB移到链表 headA 的头节点。

4、当指针 pA 指向同一个节点或者都为空时，返回它们指向的节点或者 null。

public class Solution {
    public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    //只有当链表 headA 和 headB都不为空时，两个链表才可能相交。因此首先判断链表 headA和 headB 是否为空，如果其中至少有一个链表为空，则两个链表一定不相交，返回null。
        if(headA==null||headB==null){
            return null;
        }
        //当链表 headA和 headB都不为空时，创建两个指针 pA 和 pB，初始时分别指向两个链表的头节点 headA和 headB，然后将两个指针依次遍历两个链表的每个节点。
        ListNode pA=headA,pB=headB;
        while(pA!=pB){
            pA=pA==null?headB:pA.next;
            pB=pB==null?headA:pB.next;
        }
        return pA;
    }
}

 

思路二：哈希集合       时间复杂度O(m+n)

判断两个链表是否相交，可以使用哈希集合存储链表节点。

1、首先遍历链表 headA，并将链表 headA\中的每个节点加入哈希集合中。然后遍历链表 headB，对于遍历到的每个节点，判断该节点是否在哈希集合中：

2、如果当前节点不在哈希集合中，则继续遍历下一个节点；

3、如果当前节点在哈希集合中，则后面的节点都在哈希集合中，即从当前节点开始的所有节点都在两个链表的相交部分，因此在链表 headB 中遍历到的第一个在哈希集合中的节点就是两个链表相交的节点，返回该节点。

4、如果链表 headB 中的所有节点都不在哈希集合中，则两个链表不相交，返回 null。

public class Solution {
    public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
        Set<ListNode> visited = new HashSet<ListNode>();
        ListNode temp = headA;
        while (temp != null) {
            visited.add(temp);
            temp = temp.next;
        }
        temp = headB;
        while (temp != null) {
            if (visited.contains(temp)) {
                return temp;
            }
            temp = temp.next;
        }
        return null;
    }
}