leetcode 160.相交链表 golang实现

描述

编写一个程序，找到两个单链表相交的起始节点。
如果两个链表没有交点，返回 null.
在返回结果后，两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度，且仅用 O(1) 内存

思路

如果两个链表相交 如果链表分别长度为m n 且m > n则:
让长度为m的链表先遍历到m -n 的地方 再让 链表同时遍历 那么一定能找到相交的点

实现

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
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
    
}