题目地址:
https://leetcode.com/problems/remove-duplicates-from-sorted-list/
给定一个有序链表,去除重复数字。一般而言,引入一个dummy节点会使得代码得到简化,因为有了dummy节点以后,链表头节点和后面的节点地位类似,可以统一处理。
public class Solution {
public ListNode deleteDuplicates(ListNode head) {
if (head == null) {
return head;
}
ListNode dummy = new ListNode(0);
ListNode cur = dummy;
while (head != null) {
// 只需遍历的时候考虑当前节点是否需要加入到cur的后面即可
if (head.next == null || head.val != head.next.val) {
cur.next = head;
cur = cur.next;
}
head = head.next;
}
return dummy.next;
}
}
class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。
算法正确性证明:
只需证明每次while循环内的if结束时,从dummy.next
到cur
之间已经存储了head
以及其之前的去重后的节点。这里采取的策略是,每次只加入诸多重复节点的其中的最后一个。显然“最后一个”等价于它的next为null,或者它的next的val和它自己的val不相等。所以正确性成立。举例来说,对于如下的链表
(
1
,
1
,
2
,
3
,
3
)
(1,\textbf{1},2,3,\textbf{3})
(1,1,2,3,3)加到cur后面的节点是加粗的这几个,而这几个的共同点就是,或者它的next为null,或者它自己的val和它的next的val不相等。当然也可以采取别的策略,比如每次只加重复元素的第一个,也是可以的。