LeetCode---Remove Duplicates from Sorted List II

题目链接

题意: 给出单链表的head指针, 要求去除链表中所有出现重复的元素, 如1->2->3->3->4->4->5, 返回 1->2->5

这题纠结了有两天,  重要的是把思路想清楚然后就可以痛苦的A掉, 不然老是会绕来绕去...

我的大体思路是这样的: 使用三个指针 pre保存链表中前一个没有出现重复的结点, p保存当前测试是否出现重复的结点, q用来测试p.

使用flag标记是否是第一次找到没有出现重复结点. 如果当前被测试的结点为空或者后面已经没有结点, 直接返回.

 

附上代码:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *deleteDuplicates(ListNode *head) {
        ListNode *p = head, *pre = head;
        int flag = 0;
        while (true) {
            if (p == NULL or p->next == NULL) {
                return head;
            } else {
                ListNode *q = p->next;
                while (q!= NULL and q->val == p->val) {
                    q = q->next;
                }
                if (p->next == q) {
                    if (flag == 0) {
                        flag = 1;
                        head = p;
                    } else {
                        pre = p;
                    }
                } else {
                    if (flag == 0) {
                        head = pre = q;
                    } else {
                        pre->next = q;
                    }
                }
                p = q;
            }
        }
        return head;
    }
};