删除链表中重复的节点

题目描述

在一个排序的链表中，存在重复的结点，请删除该链表中重复的结点，重复的结点不保留，返回链表头指针。例如，链表1->2->3->3->4->4->5处理后为1->2->5。

struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
        val(x), next(NULL) {
    }
};

看题目的第一眼以为很容易，做过很多删除链表结点的题目了感觉很熟悉，但是上手后发现并不容易，一般的删除重复结点的题目是要求重复的结点只保留一个，而本题要求链表中重复的节点都不保留。

思路

设置指针pre指向访问过的最后一个不重复的结点，设置工作指针p指向当前处理的结点，设置q指针检测当前结点后的重复结点。

题解

class Solution
{
public:
    ListNode *deleteDuplication(ListNode *pHead)
    {
        if (pHead == NULL)
        {
            return pHead;
        }
        ListNode *pre = NULL;           //指向访问过的最后一个不重复的结点
        ListNode *p = pHead;			//指向当前结点
        ListNode *q = NULL;				//指向当前结点后的结点，用于判断后面的结点是否重复
        while (p != NULL)
        {
        	//如果p结点为重复结点
            if (p->next != NULL && p->next->val == p->val)
            {
            	//如果当前节点后一个结点的值与当前结点重复，指针q开始工作，指向p->next
                q = p->next;
                //q一直后移直到遇到不与p结点重复的结点，循环跳出时q指向重复结点的最后一个或指向链尾。
                while (q != NULL && q->next != NULL && q->next->val == p->val)
                {
                    q = q->next;
                }
                //若头结点为重复结点，则将头指针指向重复结点后的一个。即p->...->q->pHead->...
                if (p == pHead)
                {
                    pHead = q->next;
                }
                //若头结点不是重复结点，则用pre指向重复的结点后一个结点，即pHead->...->p->...->q->pre->...
                else
                {
                    pre->next = q->next;
                }
                //更新p指针为当前重复结点的后一个结点，对该结点是否重复进行判断
                p = q->next;
            }
            //如果p结点不重复，则用pre记录，p更新，即pHead->...->pre->p->...
            else
            {
                pre = p;
                p = p->next;
            }
        }
        return pHead;
    }
};