描述
编写一个程序,找到两个单链表相交的起始节点。
如果两个链表没有交点,返回 null.
在返回结果后,两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存
思路
如果两个链表相交 如果链表分别长度为m n 且m > n则:
让长度为m的链表先遍历到m -n 的地方 再让 链表同时遍历 那么一定能找到相交的点
实现
func getIntersectionNode(headA, headB *ListNode) *ListNode {
if headA == nil || headB == nil{
return nil
}
aLen := 0
bLen := 0
curNodeA := headA
curNodeB := headB
for curNodeA != nil || curNodeB != nil{
if curNodeA != nil{
aLen++
curNodeA = curNodeA.Next
}
if curNodeB != nil{
bLen++
curNodeB = curNodeB.Next
}
}
curNodeA = headA
curNodeB = headB
var cross *ListNode
if aLen > bLen{
for i := 0; i < aLen - bLen; i++{
curNodeA = curNodeA.Next
}
}else{
for i := 0; i < bLen - aLen; i++{
curNodeB = curNodeB.Next
}
}
for curNodeB != nil && curNodeA != nil{
if curNodeA == curNodeB{
cross = curNodeB
break
}
curNodeB = curNodeB.Next
curNodeA = curNodeA.Next
}
return cross
}