给定一个已排序的链表,删除所有有重复数字的节点,只留下原始列表中不同的数字。
例子:
-
给定
1->2->3->3->4->4->5->null
,返回1->2->5->null
-
给定
1->1->1->2->3->null
,返回2->3->null
问题:
- 给定
1->1->1->2->3->null
,我的解决方案低于return 3->null
/**
* 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 the linked list
*/
public static ListNode deleteDuplicates(ListNode head) {
// write your code here
if(head == null || head.next == null) return head;
ListNode post = head.next;
ListNode curr = head;
ListNode dummy = new ListNode(head.val-1); // make sure dummy node value is different from the head
dummy.next = head;
ListNode prev = dummy;
while(post != null) {
//System.out.println("prev.val = " + prev.val + ", curr.val = " + curr.val + ", post.val = " + post.val + ", prev.next.val = " + prev.next.val);
//System.out.println("prev.next.val = " + prev.next.val + ", curr.next.val = " + curr.next.val);
if(prev.next.val == curr.val && prev.next.val == post.val) {
curr = curr.next;
post = post.next;
}
else if(prev.next.val == curr.val && prev.next.val != post.val) {
curr = curr.next;
post = post.next;
prev.next = curr;
}
else {
prev = prev.next;
curr = curr.next;
post = post.next;
}
}
return dummy.next;
}
}
不改变程序的功能,可以将while
循环转换为更易于阅读的形式:
while (post != null) {
if (prev.next.val != curr.val || prev.next.val != post.val) {
if (prev.next.val == curr.val) {
prev.next = curr.next;
} else {
prev = prev.next;
}
}
curr = curr.next;
post = post.next;
}
这相当于您的实际代码。我会根据这个版本来解释,因为我觉得这更容易阅读和推理。
让我们观察一些事情:
一开始,
prev.next
指向curr
。所以prev.next.val
等于curr.val
。此外,post
比curr
领先一步。curr
和post
一起移动。curr
和post
在if
条件下不改变;作为循环的最后一步,
给定输入1,1,1,2,3和上述观察值:
外部
if
条件将为假,直到post
达到2。curr
落后一步,所以它指向2前面的1。prev
没有移动,所以prev.next
仍然指向第一个1。所以此时,
prev.next.val
等于curr.val
(均为1),但它不等于post.val
,也就是2。所以我们输入外部if作为
prev.next.val == curr.val
,我们也进入内部if
,并设置prev.next = curr.next
。记住,循环的最后一步将把curr
推进到curr.next
。所以prev.next
将指向curr
。在下一次迭代中,我们将
post
置于3,curr
置于2,prev.next
指向curr
。因此,我们输入另一个if
,然后是内部的if
,将prev.next
设置为curr.next
,即3。
到此结束。prev
没有移动,它一直呆在原地,也就是dummy
。prev.next
指向3,我们返回的是错误的。注意,如果输入较长,例如1,1,1,2,3,4,5,6,同样的行为还会继续,curr
之后的prev.next
,实现会错误地返回6 -> null
。算法被破坏了
考虑这个替代算法:
- 创建一个虚拟节点,指向
head
作为下一个(就像你已经做的那样) - 设置当前节点为虚拟
- 下一个节点存在,下一个节点也存在
- 比较
next.val
和next.next.val
- 如果不等于,则前进当前节点
- 如果相等,则:
- 保存
next.val
的副本 - 跳过
next
和next.next
(next.next.next
旁边的设置) - 跳过所有与保存的
val
值相等的节点
- 保存
- 比较
- 返回
dummy.next
:
if (head == null) return head;
ListNode dummy = new ListNode(head.val - 1);
dummy.next = head;
ListNode node = dummy;
while (node.next != null && node.next.next != null) {
if (node.next.val != node.next.next.val) {
node = node.next;
} else {
int val = node.next.val;
node.next = node.next.next.next;
while (node.next != null && node.next.val == val) {
node.next = node.next.next;
}
}
}
return dummy.next;