leetcode 面试题02.01 移除重复节点

题目

编写代码，移除未排序链表中的重复节点。保留最开始出现的节点。
示例1:

输入：[1, 2, 3, 3, 2, 1]
输出：[1, 2, 3]

示例2:

输入：[1, 1, 1, 1, 2]
输出：[1, 2]
提示：

链表长度在[0, 20000]范围内。
链表元素在[0, 20000]范围内。
进阶：
如果不得使用临时缓冲区，该怎么解决？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-duplicate-node-lcci

分析

脑子第一个想法，果断是一个map一把撸。这样是一个空间复杂度O(N)的解法，为了更节约内存，我们使用go语言的空结构体struct{}，这样就是一个简单的set了。

解法

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func removeDuplicateNodes(head *ListNode) *ListNode {
    if head ==  nil {
        return head
    }
    var m = make(map[int]struct{})
    mHead := &ListNode{
        Next:head,
    }
    pre, cur := mHead, mHead.Next
    for cur != nil {
        if _, ok := m[cur.Val]; ok {
            pre.Next = cur.Next
        } else {
            m[cur.Val] = struct{}{}
            pre = cur
        }
        cur = cur.Next
    }

    return mHead.Next
}