题解 | 删除有序链表中重复的元素-II

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
#include <ostream>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @return ListNode类
     */
    ListNode* deleteDuplicates(ListNode* head) {
        // write code here
        ListNode* newHead = new ListNode(0);
        newHead->next = head;

        ListNode* pre = newHead;
        ListNode* p = head;

        while(p && p->next) {
            int cnt = 1;

            while (p && p->next && p->next->val == p->val) {
                cnt++;
                p->next = p->next->next;
            }

            // cout << p->val << endl;
            // cout << "cnt = " << cnt << endl;
            // 节点值出现次数 > 1.
            if (cnt > 1) {
                p = p->next;
                pre->next = p;
                continue;
            }

            p = p->next;
            pre = pre->next;

            // cout << pre->val << " " << p->val << endl; 
        }

        return newHead->next;
    }
};

用一个pre指针指向当前节点的前一个节点元素，按照上一题的方法删除与当前节点元素相等的节点

若当前节点元素出现次数大于 1 ，则将该节点一并删除

Day4