题目:
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
.
思路:从head开始遍历,当遇到重复的节点记录下首次重复的节点,当两个相邻节点值不等时判断是否有记录重复的节点,如果有则需要删除操作,只需将首次记录的节点的next指向新的节点,如此,当循环结束时再处理一次标志位。代码如下:
class Solution {
public:
ListNode *deleteDuplicates(ListNode *head) {
if(head == NULL)
return NULL;
ListNode *flag = NULL;
ListNode *temp_head = head;
while(temp_head->next){
if(temp_head->val == temp_head->next->val){
// First time repeat.
if(flag == NULL){
flag = temp_head;
}
}else{
// Delete the repeated node.
if(flag != NULL){
flag->next = temp_head->next;
flag = NULL;
}
}
temp_head = temp_head->next;
}
if(flag != NULL){
flag->next = NULL;
}
return head;
}
};
这道题的加强版:
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
.
class Solution {
public:
ListNode *deleteDuplicates(ListNode *head) {
if(head == NULL)
return NULL;
ListNode *flag = NULL;
ListNode *temp_head = head;
while(temp_head->next){
if(temp_head->val == temp_head->next->val){
// First time repeat.
if(flag == NULL){
flag = temp_head;
}
temp_head = temp_head->next;
}else{
// Delete the repeated node.
if(flag != NULL){
flag->val = temp_head->next->val;
flag->next = temp_head->next->next;
// Because use flag replace the temp_head so need to reset the temp_head;
temp_head = flag;
flag = NULL;
}else{
temp_head = temp_head->next;
}
}
}
if(flag != NULL){
if(flag == head)
head = NULL;
else{
// Search the last repeated node's previous.
temp_head = head;
while(temp_head->next != flag)
temp_head = temp_head->next;
temp_head->next = NULL;
}
}
return head;
}
};