题目
编写代码,移除未排序链表中的重复节点。保留最开始出现的节点。
示例1:
输入:[1, 2, 3, 3, 2, 1]
输出:[1, 2, 3]
示例2:
输入:[1, 1, 1, 1, 2]
输出:[1, 2]
提示:
- 链表长度在[0, 20000]范围内。
- 链表元素在[0, 20000]范围内。
进阶:
如果不得使用临时缓冲区,该怎么解决?
解题思路
遍历链表,用一个哈希集合(HashSet)来存储所有出现过的节点。
具体地,我们从链表的头节点 head 开始进行遍历,遍历的指针记为 cur。由于头节点一定不会被删除,因此我们可以枚举待删除节点的前驱节点 cur,那么当前节点就是 cur.next,后继节点就是 cur.next.next:
1)如果当前节点的值不在 Set 中,就将值加入 Set,并将指针移动到下一位置。
2)如果当前节点的值已在 Set 中出现过,那么就删除当前节点,将当前节点的后继节点与前驱节点相连。
复杂度分析:
时间复杂度:O(N),其中 N 是给定链表中节点的数目。
空间复杂度:O(N)。在最坏情况下,给定链表中每个节点都不相同,哈希表中需要存储所有的 N 个值。
代码
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode removeDuplicateNodes(ListNode head) {
if(head == null){
return head;
}
Set<Integer> set = new HashSet<>();
set.add(head.val);
ListNode cur = head;
while(cur.next != null){
if(!set.contains(cur.next.val)){
set.add(cur.next.val);
cur = cur.next;
}else{
cur.next = cur.next.next;
}
}
return head;
}
}