Given a sorted linked list, delete all nodes that have duplicate numbers, leaving only distinct numbers from the original list.
For example,
Given 1->2->3->3->4->4->5
, return 1->2->5
.
Given 1->1->1->2->3
, return 2->3
.
这是上一问题的升级版。开始做时有些迷茫,想着如果是三个元素连续时的情况怎么处理,还分了奇偶进行讨论,但是很复杂,还要考虑剩两个元素时怎么处理等等~后来看到别人的这个,用的递归,思路比较清晰,学习记录一下。
/**
* 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 || !head->next)
return head;//当链表为空或者为一个元素时直接返回
ListNode *p = head->next;
if (head->val == p->val)
{
while (p && head->val == p->val)//这个while非常好
{
ListNode *tmp = p;
p = p->next;
delete tmp;
}
delete head;//注意这里的delete是while不再执行时进行的,所以在此之前head一直能够与后面的元素比较
return deleteDuplicates(p);//此时上一个重复元素的结点已经全部删除,递归进行下一元素判断
}
else
{
head->next = deleteDuplicates(head->next);//如果前面结点没有重复元素,则递归对下一结点进行判定
return head;
}
}
};