160. 相交链表

题目描述

思路
解决这道题最快的思路其实是首先对headA所在的链表进行一次遍历,使用一个Map来模拟Set记录出现过的节点,之后再对headB所在的链表进行一次遍历,如果其中有节点的递增与Map当中保存的地址相同,那么这个节点就是两个链表相交的节点,直接返回该节点即可;否则,就代表不存在一个节点使两个链表相交。
我们不借助额外的数据结构,仅基于链表的长度,使用另一个思路来解决这个问题。
具体来说,我们先对这两个链表进行一次遍历,得到lenA和lenB,即两个链表的长度。我们需要确保lenA < lenB,因此如果lenA > lenB,则交换两个链表以及对应的长度值。
再然后,现在我们已知headA所在的链表短于headB所在的链表,我们令p, q := headA, headB,让q先前进lenB - lenA,然后再同时移动p和q直到两个指针相遇,相遇的节点就是两个链表相交的节点,直接返回即可。
基于以上思路,我们写代码来解决问题。
Golang 题解
go
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func getIntersectionNode(headA, headB *ListNode) *ListNode {
lenA, lenB := 0, 0
p, q := headA, headB
for p != nil {
lenA ++
p = p.Next
}
for q != nil {
lenB ++
q = q.Next
}
if lenA > lenB {
lenA, lenB, headA, headB = lenB, lenA, headB, headA
}
p, q = headA, headB
for i := 0; i < lenB - lenA; i ++ {
q = q.Next
}
for p != q && p != nil{
p = p.Next
q = q.Next
}
return p
}