160. 相交鏈表
編寫一個(gè)程序，找到兩個(gè)單鏈表相交的起始節(jié)點(diǎn)。
如下面的兩個(gè)鏈表：
image.png
在節(jié)點(diǎn) c1 開始相交。
示例 1：
image.png
輸入：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
輸入解釋：相交節(jié)點(diǎn)的值為 8 （注意，如果兩個(gè)列表相交則不能為 0）。從各自的表頭開始算起，鏈表 A 為 [4,1,8,4,5]，鏈表 B 為 [5,0,1,8,4,5]。在 A 中，相交節(jié)點(diǎn)前有 2 個(gè)節(jié)點(diǎn)；在 B 中，相交節(jié)點(diǎn)前有 3 個(gè)節(jié)點(diǎn)。
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func getIntersectionNode(headA, headB *ListNode) *ListNode {
var listal, listbl int
headA2, headB2 := headA, headB
for headA2 != nil {
listal++
headA2 = headA2.Next
}
for headB2 != nil {
listbl++
headB2 = headB2.Next
}
if listal > listbl {
for i := 1; i <= listal-listbl; i++ {
headA = headA.Next
}
} else if listal < listbl {
for i := 1; i <= listbl-listal; i++ {
headB = headB.Next
}
}
for headA != nil && headB != nil {
if headA == headB {
return headA
}
headA = headA.Next
headB = headB.Next
}
return nil
}
