【程序员面试金典】02.01. 移除重复节点(set)

1.题目

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

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

2.题解

class Solution
{
public:
    ListNode* removeDuplicateNodes(ListNode* head)
    {
        if(head==NULL) return head;
        unordered_set<int> s;
        ListNode* p=head;
        s.insert(p->val);
        while(p->next!=NULL) //
        {
            if(s.count(p->next->val)) //有
            {
                p->next=p->next->next;
            }
            else
            {
                s.insert(p->next->val);//沒有
                p=p->next;
            }
        }
        return head;
    }
};

不使用缓冲区：

class Solution
{
public:
    ListNode* removeDuplicateNodes(ListNode* head)
    {
        if(head == NULL || head->next == NULL)
        {
            return head;
        }
        ListNode* p1 = head;
        ListNode* p2 = NULL;
        ListNode* tmp = NULL;
        while(p1 != NULL)
        {
            tmp = p1;
            p2 = p1->next;
            while(p2 != NULL)
            {
                if(p1->val == p2->val)
                {
                    tmp->next = p2->next;
                }
                else
                {
                    tmp = p2;
                }
                p2 = tmp->next;
            }
            p1 = p1->next;
        }
        return head;
    }
};