欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【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)