给你两个单链表的头节点 headA
和 headB
,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null
。
这道题自己解的时候是最笨的思路,先分别求出两个链表的长度,以短链表的长度作为循环的结束条件,以此与另一个链表比较,相等则返回,不相等长链表到末尾,短链表下移。最笨的方法,所以提交的效率也很低,直接见代码
func getIntersectionNode(headA, headB *ListNode) *ListNode {
if headA == nil || headB == nil{
return nil
}
p := headA
q := headB
lenA := 0
lenB := 0
//得到两个链表的长度
if p != nil || q != nil{
if p!= nil{
lenA++
p =p.Next
}
if q!= nil{
lenB++
q=q.Next
}
}
p, q = headA, headB
//保证A是短链表,也就是p指向短链表
if lenA > lenB{
p = headB
q = headA
t := lenA
lenA = lenB
lenB = t
}
count := 0
for count < lenA && p != nil{
t := q
for t != nil {
if p == t{
return p
}
t = t.Next
}
p = p.Next
}
return nil
}
看了官方解题思路后,感觉这种方法比较靠谱,同时利用了map和访问的属性,直接上代码
func getIntersectionNode(headA, headB *ListNode) *ListNode {
if headA == nil || headB == nil{
return nil
}
vis := map[*ListNode]bool{}
for p:=headA;p!=nil;p=p.Next{
vis[p] = true
}
for q := headB;q!=nil;q=q.Next{
if vis[q] == true{
return q
}
}
return nil
}