力扣82-删除链表中的重复元素II-C++

一、运行结果

 

二、题目

给定一个已排序的链表的头 head ， 删除原始链表中所有重复数字的节点，只留下不同的数字 。返回 已排序的链表 。

示例 1：
输入：head = [1,2,3,3,4,4,5]
输出：[1,2,5]
示例 2：
输入：head = [1,1,1,2,3]
输出：[2,3]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-list-ii

三、思路

鉴于链表已经有序，设置三个指针：q 指向新链表的最后一个元素，pre 指向该val值的第一个结点，p 指向和pre 值相同的最后一个节点的下一个结点，用pre 和 p 遍历原链表， 如[1,2,2,2,3]，当pre指向第一个2 时，先指向pre的下一个结点，如果p->val = pre->val，p就一直后移，直到 p 的值为3，如果退出循环时 p 仍然执行pre 的下一个节点，那么没有和pre 值相同的结点，将pre 插入到q 的后面，q, pre, p 都向后走一步，继续遍历；否则pre 到 p 的前驱节点的值都是相同的，跳过这一段，令pre = p， p = p->next，继续向后遍历。

四、代码

class Solution {
public:
    ListNode* deleteDuplicates(ListNode* head) {
        if(head == NULL || head->next == NULL) return head;
        ListNode* dummy = new ListNode(-1);
        dummy->next = head;
        ListNode* q = dummy;  //q指向新链表的最后一个节点
        ListNode* pre = head;
        ListNode* p = head->next;
        while(p != NULL){
            while(p != NULL && p->val == pre->val){
                p = p->next;
            }
            if(p == pre->next){  //指向相邻两节点，非重复元素
                q->next = pre;
                q = q->next;
                pre = pre->next;
                p = p->next;
            }
            else if(p != NULL){
                pre = p;
                p = p->next;
            }
            else{
                q->next = p;
            }
        }
        if(pre->next == NULL){  //pre指向最后一个节点
            q->next = pre;
        }
        return dummy->next;
    }
};