和string中双指针算法很类似,index i
和j
换成了指针p
和q
。
C++ 代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
// [p, q)之间为重复元素,t = p - 1,如下所示
// 1 2 3 3 3 4 5
// t p q
ListNode* deleteDuplication(ListNode* head) {
ListNode *p = head;
ListNode *t = NULL;
while (p != NULL) {
int cnt = -1;
ListNode *q = p;
while (q != NULL && q->val == p->val) {
q = q->next;
++cnt;
}
if (cnt == 0) { // 无重复数字
t = p;
p = p->next;
} else { // 有重复数字
if (t == NULL) // 开头元素重复,需要更新head
head = q;
else
t->next = q; // 删除[p, q)之间的重复数字
p = q;
}
}
return head;
}
};