题目让从一个排好序的链表中删除重复的元素,使得重复的元素一次也不出现。
思路是采取差一法。
差一法:即要删除一个链表节点的时候,往往需要保持这个节点的前置节点,所以在判断要删除节点的时候直接采取node.next.val与目标值进行比较,而不是用node.val和目标值进行比较,也就是用下一个节点的值和目标值比较。这种方法叫做差一法。
这道题和I不同点在于第一个元素有可能会被删掉,所以必须构造一个helper来辅助删除第一个元素。
先给出完整代码,再重点分析部分代码
public ListNode deleteDuplicates(ListNode head) {
if (head == null)
return head;
ListNode helper = new ListNode(0);
helper.next = head;
ListNode pre = helper;
ListNode cur = head;
while (cur != null) {
while (cur.next != null && pre.next.val == cur.next.val)
cur = cur.next;
if (pre.next == cur) {
pre = pre.next;
cur = cur.next;
} else {
pre.next = cur.next;
cur = pre.next;
}
}
return helper.next;
}
这里有三个重点部分需要大家理解
第一个部分就是差一法的比较思想,如果遇见了相等的元素,那么当前指针继续移动,直到元素不再相等的时候结束。
while (cur.next != null && pre.next.val == cur.next.val)
cur = cur.next;
第二个部分是如果两个元素不相等,说明cur和pre之间的差距还是1,所以不用删除,只需更新pre和cur指针即可。
if (pre.next == cur) {
pre = pre.next;
cur = cur.next;
}
第三部分是如果两个元素相等了,那么需要删除这些相等的所有元素,只更新cur指针即可。千万不要更新pre指针,这样会出错。
比如0-1-1-2-2-3
当你删掉了1-1的时候,链表为0-2-2-3,这个时候pre指向0,cur指向第一个2,所以pre不能更新,不然2-2就会被跳过。
else {
pre.next = cur.next;
cur = pre.next;
}