Question

1429. 第一个唯一数字

English Version

题目描述

给定一系列整数，插入一个队列中，找出队列中第一个唯一整数。

实现 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

 

提示：

• 1 <= nums.length <= 10^5
• 1 <= nums[i] <= 10^8
• 1 <= value <= 10^8
• 最多调用 5000 次 showFirstUnique 和 add 。

解法

方法一：哈希表 + 双端队列

我们可以使用哈希表 $cnt$ 统计每个数字出现的次数，使用双端队列 $q$ 按顺序维护出现的数字。

调用 showFirstUnique 方法时，判断队列 $q$ 的队头元素是否在哈希表 $cnt$ 中出现的次数是否为 $1$，如果是，则返回队头元素，否则将队头元素弹出，直到队列为空或者队头元素在哈希表 $cnt$ 中出现的次数为 $1$，如果队列为空，则返回 $-1$。

调用 add 方法时，将数字加入哈希表 $cnt$ 中，并将数字加入队列 $q$ 中。

时间复杂度 $O(n)$，空间复杂度 $O(n)$。其中 $n$ 为数组 $nums$ 的长度。

class FirstUnique:
  def __init__(self, nums: List[int]):
    self.cnt = Counter(nums)
    self.unique = OrderedDict({v: 1 for v in nums if self.cnt[v] == 1})

  def showFirstUnique(self) -> int:
    return -1 if not self.unique else next(v for v in self.unique.keys())

  def add(self, value: int) -> None:
    self.cnt[value] += 1
    if self.cnt[value] == 1:
      self.unique[value] = 1
    elif value in self.unique:
      self.unique.pop(value)


# Your FirstUnique object will be instantiated and called as such:
# obj = FirstUnique(nums)
# param_1 = obj.showFirstUnique()
# obj.add(value)

class FirstUnique {
  private Map<Integer, Integer> cnt = new HashMap<>();
  private Set<Integer> unique = new LinkedHashSet<>();

  public FirstUnique(int[] nums) {
    for (int v : nums) {
      cnt.put(v, cnt.getOrDefault(v, 0) + 1);
    }
    for (int v : nums) {
      if (cnt.get(v) == 1) {
        unique.add(v);
      }
    }
  }

  public int showFirstUnique() {
    return unique.isEmpty() ? -1 : unique.iterator().next();
  }

  public void add(int value) {
    cnt.put(value, cnt.getOrDefault(value, 0) + 1);
    if (cnt.get(value) == 1) {
      unique.add(value);
    } else {
      unique.remove(value);
    }
  }
}

/**
 * Your FirstUnique object will be instantiated and called as such:
 * FirstUnique obj = new FirstUnique(nums);
 * int param_1 = obj.showFirstUnique();
 * obj.add(value);
 */

class FirstUnique {
public:
  FirstUnique(vector<int>& nums) {
    for (int& v : nums) {
      ++cnt[v];
      q.push_back(v);
    }
  }

  int showFirstUnique() {
    while (q.size() && cnt[q.front()] != 1) q.pop_front();
    return q.size() ? q.front() : -1;
  }

  void add(int value) {
    ++cnt[value];
    q.push_back(value);
  }

private:
  unordered_map<int, int> cnt;
  deque<int> q;
};

/**
 * Your FirstUnique object will be instantiated and called as such:
 * FirstUnique* obj = new FirstUnique(nums);
 * int param_1 = obj->showFirstUnique();
 * obj->add(value);
 */

type FirstUnique struct {
  cnt map[int]int
  q   []int
}

func Constructor(nums []int) FirstUnique {
  cnt := map[int]int{}
  for _, v := range nums {
    cnt[v]++
  }
  return FirstUnique{cnt, nums}
}

func (this *FirstUnique) ShowFirstUnique() int {
  for len(this.q) > 0 && this.cnt[this.q[0]] != 1 {
    this.q = this.q[1:]
  }
  if len(this.q) > 0 {
    return this.q[0]
  }
  return -1
}

func (this *FirstUnique) Add(value int) {
  this.cnt[value]++
  this.q = append(this.q, value)
}

/**
 * Your FirstUnique object will be instantiated and called as such:
 * obj := Constructor(nums);
 * param_1 := obj.ShowFirstUnique();
 * obj.Add(value);
 */

方法二

class FirstUnique:
  def __init__(self, nums: List[int]):
    self.cnt = Counter(nums)
    self.q = deque(nums)

  def showFirstUnique(self) -> int:
    while self.q and self.cnt[self.q[0]] != 1:
      self.q.popleft()
    return -1 if not self.q else self.q[0]

  def add(self, value: int) -> None:
    self.cnt[value] += 1
    self.q.append(value)


# Your FirstUnique object will be instantiated and called as such:
# obj = FirstUnique(nums)
# param_1 = obj.showFirstUnique()
# obj.add(value)

class FirstUnique {
  private Map<Integer, Integer> cnt = new HashMap<>();
  private Deque<Integer> q = new ArrayDeque<>();

  public FirstUnique(int[] nums) {
    for (int v : nums) {
      cnt.put(v, cnt.getOrDefault(v, 0) + 1);
      q.offer(v);
    }
  }

  public int showFirstUnique() {
    while (!q.isEmpty() && cnt.get(q.peekFirst()) != 1) {
      q.poll();
    }
    return q.isEmpty() ? -1 : q.peekFirst();
  }

  public void add(int value) {
    cnt.put(value, cnt.getOrDefault(value, 0) + 1);
    q.offer(value);
  }
}

/**
 * Your FirstUnique object will be instantiated and called as such:
 * FirstUnique obj = new FirstUnique(nums);
 * int param_1 = obj.showFirstUnique();
 * obj.add(value);
 */