Question

1481. 不同整数的最少数目

English Version

题目描述

给你一个整数数组 arr 和一个整数 k 。现需要从数组中恰好移除 k 个元素，请找出移除后数组中不同整数的最少数目。

 

示例 1：

输入：arr = [5,5,4], k = 1
输出：1
解释：移除 1 个 4 ，数组中只剩下 5 一种整数。

示例 2：

输入：arr = [4,3,1,1,3,3,2], k = 3
输出：2
解释：先移除 4、2 ，然后再移除两个 1 中的任意 1 个或者三个 3 中的任意 1 个，最后剩下 1 和 3 两种整数。

 

提示：

• 1 <= arr.length <= 10^5
• 1 <= arr[i] <= 10^9
• 0 <= k <= arr.length

解法

方法一：哈希表 + 排序

我们用哈希表 $cnt$ 统计数组 $arr$ 中每个整数出现的次数，然后将 $cnt$ 中的值按照从小到大的顺序排序，记录在数组 $nums$ 中。

接下来，我们遍历数组 $nums$，对于当前遍历到的每个值 $nums[i]$，我们将 $k$ 减去 $nums[i]$，如果 $k \lt 0$，则说明我们已经移除了 $k$ 个元素，此时数组中不同整数的最少数目为 $nums$ 的长度减去当前遍历到的下标 $i$，直接返回即可。

若遍历结束，说明我们移除了所有的元素，此时数组中不同整数的最少数目为 $0$。

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

class Solution:
  def findLeastNumOfUniqueInts(self, arr: List[int], k: int) -> int:
    cnt = Counter(arr)
    for i, v in enumerate(sorted(cnt.values())):
      k -= v
      if k < 0:
        return len(cnt) - i
    return 0

class Solution {
  public int findLeastNumOfUniqueInts(int[] arr, int k) {
    Map<Integer, Integer> cnt = new HashMap<>();
    for (int x : arr) {
      cnt.merge(x, 1, Integer::sum);
    }
    List<Integer> nums = new ArrayList<>(cnt.values());
    Collections.sort(nums);
    for (int i = 0, m = nums.size(); i < m; ++i) {
      k -= nums.get(i);
      if (k < 0) {
        return m - i;
      }
    }
    return 0;
  }
}

class Solution {
public:
  int findLeastNumOfUniqueInts(vector<int>& arr, int k) {
    unordered_map<int, int> cnt;
    for (int& x : arr) {
      ++cnt[x];
    }
    vector<int> nums;
    for (auto& [_, c] : cnt) {
      nums.push_back(c);
    }
    sort(nums.begin(), nums.end());
    for (int i = 0, m = nums.size(); i < m; ++i) {
      k -= nums[i];
      if (k < 0) {
        return m - i;
      }
    }
    return 0;
  }
};

func findLeastNumOfUniqueInts(arr []int, k int) int {
  cnt := map[int]int{}
  for _, x := range arr {
    cnt[x]++
  }
  nums := make([]int, 0, len(cnt))
  for _, v := range cnt {
    nums = append(nums, v)
  }
  sort.Ints(nums)
  for i, v := range nums {
    k -= v
    if k < 0 {
      return len(nums) - i
    }
  }
  return 0
}

function findLeastNumOfUniqueInts(arr: number[], k: number): number {
  const cnt: Map<number, number> = new Map();
  for (const x of arr) {
    cnt.set(x, (cnt.get(x) || 0) + 1);
  }
  const nums: number[] = [];
  for (const [_, v] of cnt) {
    nums.push(v);
  }
  nums.sort((a, b) => a - b);
  for (let i = 0; i < nums.length; ++i) {
    k -= nums[i];
    if (k < 0) {
      return nums.length - i;
    }
  }
  return 0;
}