[LeetCode](面试题 02.01)移除重复节点

题目

编写代码,移除未排序链表中的重复节点。保留最开始出现的节点。

示例1:

 输入:[1, 2, 3, 3, 2, 1]
 输出:[1, 2, 3]

示例2:

 输入:[1, 1, 1, 1, 2]
 输出:[1, 2]

提示:

  1. 链表长度在[0, 20000]范围内。
  2. 链表元素在[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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章