删除链表中重复的节点

ryluo 2020-06-24 21:20:48

在一个排序的链表中,存在重复的结点,请删除该链表中重复的结点,重复的结点不保留。

样例1

输入:1->2->3->3->4->4->5

输出:1->2->5

样例2

输入:1->1->1->2->3

输出:2->3

分析:

这道题是删除所有重复的元素,并不是常见的保留一个的情况。

image-20200624221042309

注意点:

对于可能对头节点进行修改的情况,可以开辟一个虚拟头节点。

代码中q至少移动一次。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* deleteDuplication(ListNode* head) {
        // 凡是头节点可能被删除掉的情况使用虚拟头节点会省很多事
        auto dummy = new ListNode(-1);
        dummy->next = head;

        auto p = dummy;
        while(p->next){
            auto q = p->next;
            while(q && p->next->val == q->val) q = q->next;  // 因为初始化的时候p->next = q,所以至少会移动一次

            // 判断此段没有重复元素,也就是q只移动了一次
            if(p->next->next == q) p = p->next;
            else p->next = q;
        }

        return dummy->next;
    }
};