給定一個已排序鏈表,刪除所有重復元素讓每個元素只出現一次。
例如:
給定 1->1->2, 返回 1->2。
給定 1->1->2->3->3, 返回 1->2->3。
Given a sorted linked list, delete all duplicates such that each element appear only once.
For example,
Given 1->1->2, return 1->2.
Given 1->1->2->3->3, return 1->2->3.
切入正題,首先我們應該先判斷是否為空:
if (head == NULL) return NULL;
if (head->next == NULL) return head;
但是這兩行代碼是可以簡寫的:
if (head == NULL || head->next == NULL) return head;
下面是一個循環語句:
while (head->next) {
if (head->val == head->next->val) {
head->next = head->next->next;
}
else {
head = head->next;
}
}
意思非常簡單,首先判斷當前的節點的值與下一個節點的值是否相等,如果相等,則將下下一個節點賦值給下一個節點。
1——1——2——3——3
改寫成:
1——2——3——3
此時
while (head->next) {
if (head->val == head->next->val) {
head->next = head->next->next;
head = head->next;
}
else {
head = head->next;
}
}
理由很簡單,例如:
1——1——1——2——3——3
經過第一次變換後:
1——1——2——3——3
如果貿然將head移動到下一個節點上,那麼下一次的對比就是對比1和2了,顯然是不合理的。
到了這裡移除操作就已經完成了,但是能夠直接
所以應該在
ListNode* newHead = head;
while(){}
return newHead;
/**
* 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* newNode = head;
while (head->next) {
if (head->val == head->next->val)
head->next = head->next->next;
else
head = head->next;
}
return newNode;
}
};