【Leetcode】1338. Reduce Array Size to The Half
程序员文章站
2022-05-20 19:10:03
...
题目地址:
https://leetcode.com/problems/reduce-array-size-to-the-half/
给定一个长 n n n的数组,可以选择若干个不同数字清除(每次会把该数在数组中的所有出现都清除),问至少选多少个数字清除,可以使得被清除的数字个数不少于 n / 2 n/2 n/2。
先统计一下每个数字出现次数,然后按次数从大到小排序,看依次清除出现次数最多的数,清除多少个就能恰好达到 n / 2 n/2 n/2的总数。代码如下:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution {
public int minSetSize(int[] A) {
List<int[]> list = new ArrayList<>();
Arrays.sort(A);
for (int i = 0; i < A.length; i++) {
int j = i;
while (j < A.length && A[j] == A[i]) {
j++;
}
list.add(new int[]{A[i], j - i});
i = j - 1;
}
list.sort((x, y) -> -Integer.compare(x[1], y[1]));
int cnt = 0, res = 0;
for (int[] p : list) {
cnt += p[1];
res++;
if (cnt >= A.length / 2) {
return res;
}
}
return 0;
}
}
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间 O ( n ) O(n) O(n)。
下一篇: 笑话集原创笑话精品展第八十三期