【队列】LeetCode 1429. 第一个唯一数字

题目链接

1429. 第一个唯一数字

给定一系列整数，插入一个队列中，找出队列中第一个唯一整数。
实现 FirstUnique 类：
FirstUnique(int[] nums) 用数组里的数字初始化队列。
int showFirstUnique() 返回队列中的 第一个唯一 整数的值。如果没有唯一整数，返回 -1。（译者注：此方法不移除队列中的任何元素）
void add(int value) 将 value 插入队列中。
示例 1：
输入：
["FirstUnique","showFirstUnique","add","showFirstUnique","add","showFirstUnique","add","showFirstUnique"]
[[[2,3,5]],[],[5],[],[2],[],[3],[]]
输出：
[null,2,null,2,null,3,null,-1]
解释：
FirstUnique firstUnique = new FirstUnique([2,3,5]);
firstUnique.showFirstUnique(); // 返回 2
firstUnique.add(5); // 此时队列为 [2,3,5,5]
firstUnique.showFirstUnique(); // 返回 2
firstUnique.add(2); // 此时队列为 [2,3,5,5,2]
firstUnique.showFirstUnique(); // 返回 3
firstUnique.add(3); // 此时队列为 [2,3,5,5,2,3]
firstUnique.showFirstUnique(); // 返回 -1
示例 2：
输入：
["FirstUnique","showFirstUnique","add","add","add","add","add","showFirstUnique"]
[[[7,7,7,7,7,7]],[],[7],[3],[3],[7],[17],[]]
输出：
[null,-1,null,null,null,null,null,17]
解释：
FirstUnique firstUnique = new FirstUnique([7,7,7,7,7,7]);
firstUnique.showFirstUnique(); // 返回 -1
firstUnique.add(7); // 此时队列为 [7,7,7,7,7,7,7]
firstUnique.add(3); // 此时队列为 [7,7,7,7,7,7,7,3]
firstUnique.add(3); // 此时队列为 [7,7,7,7,7,7,7,3,3]
firstUnique.add(7); // 此时队列为 [7,7,7,7,7,7,7,3,3,7]
firstUnique.add(17); // 此时队列为 [7,7,7,7,7,7,7,3,3,7,17]
firstUnique.showFirstUnique(); // 返回 17
示例 3：
输入：
["FirstUnique","showFirstUnique","add","showFirstUnique"]
[[[809]],[],[809],[]]
输出：
[null,809,null,-1]
解释：
FirstUnique firstUnique = new FirstUnique([809]);
firstUnique.showFirstUnique(); // 返回 809
firstUnique.add(809); // 此时队列为 [809,809]
firstUnique.showFirstUnique(); // 返回 -1

思路

代码

class FirstUnique {
    private HashSet<Integer> unique = new HashSet<>();
    private HashSet<Integer> repeated = new HashSet<>();
    private List<Integer> nums = new ArrayList<>();
    private int offset = 0;

    public FirstUnique(int[] nums) {
        for(int num : nums){
            this.add(num);
        }
    }

    public int showFirstUnique() {
        //没有unique的元素，直接返回-1
        if(unique.isEmpty()){
            return -1;
        }
        //从offset开始遍历数组，找数组中第一个unique的元素
        for(int i = offset; i < nums.size(); ++i){
            if(unique.contains(nums.get(i))){
                offset = i;
                return nums.get(i);
            }
        }

        return -1;
    }

    public void add(int value) {
        //如果重复，从unique中移除，加入repeated
        if(unique.contains(value)){
            unique.remove(value);
            repeated.add(value);
        }else{
            //如果不是repeated元素且第一次出现，加入unique
            if(!repeated.contains(value)){
                unique.add(value);
                //加入数组
                nums.add(value);
            }
        }
    }
}