算法题
链表(linked list)
链表相交

链表相交

题目

给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null 。

示例 1:

输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Intersected at '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
输出:Intersected at '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 。

思路

简单来说,就是求两个链表交点节点的指针。交点不是数值相等,而是指针相等。

我们求出两个链表的长度,并求出两个链表长度的差值,然后让cur A移动到和 curB 末尾对齐的位置

此时我们就可以比较curA和curB是否相同,如果不相同,同时向后移动curA和curB,如果遇到curA == curB,则找到交点。

解法

func getIntersectionNode(headA, headB *ListNode) *ListNode {
    curA, curB := headA, headB
    lenA, lenB := 0, 0
    // 求A、B链表长度
    for curA != nil {
        lenA++
        curA = curA.Next
    }
    for curB != nil {
        lenB++
        curB = curB.Next
    }
    curA, curB = headA, headB
    // 让curA和curB对齐
    if lenA > lenB {
        for i := 0; i < lenA - lenB; i++ {
            curA = curA.Next
        }
    } else {
        for i := 0; i < lenB - lenA; i++ {
            curB = curB.Next
        }
    }
    // 比较curA和curB是否相同
    for curA != nil && curB != nil {
        if curA == curB {
            return curA
        }
        curA = curA.Next
        curB = curB.Next
    }
    return nil
}

双指针解法:

func getIntersectionNode(headA, headB *ListNode) *ListNode {
    curA, curB := headA, headB
    // 如果curA和curB不相等,则继续循环
    for curA != curB {
        // 如果curA为空,则curA指向headB,否则指向curA的下一个节点
        if curA != nil {
            curA = curA.Next
        } else {
            curA = headB
        }
        // 如果curB为空,则curB指向headA,否则指向curB的下一个节点
        if curB != nil {
            curB = curB.Next
        } else {
            curB = headA
        }
    }
    return curA
}