链表相交
题目
给你两个单链表的头节点 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
}