LeetCode - 160. Intersection of Two Linked Lists

原题链接：https://leetcode.com/problems/intersection-of-two-linked-lists/

简单题拿golang练练手。

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func getIntersectionNode(headA, headB *ListNode) *ListNode {
    
    la := getLen(headA)
    lb := getLen(headB)
    
    if(la > lb) {
        headA = skipNodes(la - lb, headA)
    } else if (lb > la) {
        headB = skipNodes(lb - la, headB)
    }
    
    for headA != nil && headB != nil {
        if(headA == headB) {
            return headA
        }
        headA = headA.Next
        headB = headB.Next
    }
    
    return nil
}

func getLen(n *ListNode) int {
    length := 0
    for n != nil {
        n = n.Next
        length ++
    }
    return length
}

func skipNodes(len int, n *ListNode) *ListNode {
    for len > 0 {
        len --
        n = n.Next
    }
    return n
}

题目很简单，找到共同的交集，返回交集的第一个node。题目似乎简化了，这里AC的提交并没有判断后续是否均相同，而是在尾对齐后直接找第一个相同的节点，就返回了。

// 第一个case中如果用值判断的话，结果肯定是1
[4,1,8,4,5] 
[5,0,1,8,4,5]

注意是相同的node，而不是val相同。题目中并没有明确说明，我也因此WA了一次。

有疑问加站长微信联系（非本文作者）