面试题 02.01. 移除重复节点

Posted 2020-12-13 xgbt

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

示例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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeDuplicateNodes(ListNode* head) {
        if (!head){
            return nullptr;
        }

        unordered_set <int> m = {head->val};

        ListNode* tmp = head;
        while (tmp->next != nullptr){
            if (!m.count(tmp->next->val)){
                m.insert(tmp->next->val);
                tmp = tmp->next;
            }
            else{
                tmp->next = tmp->next->next;
            }
        }
        tmp->next = nullptr;

        return head;
    }
};