Remove Duplicates from Sorted List
Question
- leetcode: Remove Duplicates from Sorted List
- lintcode: Remove Duplicates from Sorted List
Problem Statement
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
.
题解
遍历之,遇到当前节点和下一节点的值相同时,删除下一节点,并将当前节点next
值指向下一个节点的next
, 当前节点首先保持不变,直到相邻节点的值不等时才移动到下一节点。
Python
"""
Definition of ListNode
class ListNode(object):
def __init__(self, val, next=None):
self.val = val
self.next = next
"""
class Solution:
"""
@param head: A ListNode
@return: A ListNode
"""
def deleteDuplicates(self, head):
curt = head
while curt:
while curt.next and curt.next.val == curt.val:
curt.next = curt.next.next
curt = curt.next
return head
C++
/**
* Definition of ListNode
* class ListNode {
* public:
* int val;
* ListNode *next;
* ListNode(int val) {
* this->val = val;
* this->next = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param head: The first node of linked list.
* @return: head node
*/
ListNode *deleteDuplicates(ListNode *head) {
ListNode *curr = head;
while (curr != NULL) {
while (curr->next != NULL && curr->val == curr->next->val) {
ListNode *temp = curr->next;
curr->next = curr->next->next;
delete(temp);
temp = NULL;
}
curr = curr->next;
}
return head;
}
};
Java
/**
* Definition for ListNode
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) {
* val = x;
* next = null;
* }
* }
*/
public class Solution {
/**
* @param ListNode head is the head of the linked list
* @return: ListNode head of linked list
*/
public static ListNode deleteDuplicates(ListNode head) {
ListNode curr = head;
while (curr != null) {
while (curr.next != null && curr.val == curr.next.val) {
curr.next = curr.next.next;
}
curr = curr.next;
}
return head;
}
}
源码分析
- 首先进行异常处理,判断head是否为NULL
- 遍历链表,
curr->val == curr->next->val
时,保存curr->next
,便于后面释放内存(非C/C++无需手动管理内存) - 不相等时移动当前节点至下一节点,注意这个步骤必须包含在
else
中,否则逻辑较为复杂
使用双重while
循环处也可使用curr != null && curr.next != null
, 这样就不用单独判断head
是否为空了,但是这样会降低遍历的效率,因为需要判断两处。while
循环可只在内循环处判断,避免了冗余的判断,谢谢 @xuewei4d 提供的思路。
复杂度分析
遍历链表一次,时间复杂度为 , 使用了一个中间变量进行遍历,空间复杂度为 .