注意链表的尾部

今天做了Leetcode上面一道题Remove Duplicates from Sorted List II，要去去除链表中重复的节点，代码如下

/**
 * 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) {
        if(head==NULL||head->next==NULL)
        {
            return head;
        }
        ListNode *p=head,*prev=new ListNode(0),*head2=prev;
        int count=0;
        while(p->next!=NULL)
        {
            if(count==0&&p->val!=p->next->val)
            {
                prev->next=p;
                prev=p;
            }
            else if(p->val==p->next->val)
            {
                count++;
            }
            else if(count>0)
            {
                count=0; 
            }
            p=p->next;
        }
        if(count==0)
        {
            prev->next=p;
        }
       // else
       //{
       //     prev->next=NULL;
       //}
        head=head2->next;
        delete head2;
        return head;
    }
};

注意到39-42行被注释掉的代码，注释掉，会怎么样？

分析一下算法，count表示前面与当前指针val相等的节点个数，对于末尾指针，前面如果没有与其数值相等，则加入链表，否则，则不加入链表，但是，如果末尾指针前面存在与其相等的数，是不是可以不用管它？不是的，因为是原地对链表操作，所以如果不处理的画，那么最后prev的next还是会指向另一个节点，而这个节点并不是我们想要的，所以39-42行必须将prev的next指向空指针。当然对于存在重复的节点，可以直接delete之，如果该节点是分配到堆上的话。