02.01. 移除重复节点

02.01. 移除重复节点

1、题目

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

1）示例1

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

2）示例2

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

2、初步作答

2.1 思路

• 创建集合存储节点中首次出现的整数
• 比较节点数值是否已存在于集合中，已存在直接删除节点，不存在直接存入集合

2.2 做法

• 判断链表头结点是否为空，为空直接返回头结点
• 创建一个 Set 集合
• 比较节点数值是否已存在于集合中，已存在直接删除节点，不存在直接存入集合
• 返回头结点

2.3 代码

/**
 * 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> a = new HashSet<Integer>();
        a.add(head.val);
        ListNode list = head;
        while(list.next != null){
            if(a.add(list.next.val)){
                list = list.next;
            }else{
                list.next = list.next.next;
            }
        }
        list.next = null;
        return head;
    }
}

2.4 思考

因为链表的这道题特地去看了一下集合还有链表，发现集合真香

3、其余解法

• 来自力扣作者：jyd

class Solution {
    public ListNode removeDuplicateNodes(ListNode head) {
        ListNode pre = null, cur = head;
        HashSet<Integer> visited = new HashSet<>();
        while (cur != null) {
            if (visited.contains(cur.val)) {
                pre.next = cur.next;
            }
            else {
                visited.add(cur.val);
                pre = cur;
            }
            cur = cur.next;
        }
        return head;
    }
}