leetcode 1833. Maximum Ice Cream Bars(最大数量的冰棒)

leetcode 1833. Maximum Ice Cream Bars(最大数量的冰棒)_第1张图片
数组costs里面是每根冰棒的价格,现手上有coins的钱,问最多能买多少个冰棒。

思路:

最容易想到的就是给价格从小到大排个序,先买便宜的再买贵的,就能保证买的最多。

    public int maxIceCream(int[] costs, int coins) {
        int n = costs.length;
        int res = 0;

        Arrays.sort(costs);

        for(int i = 0; i < n; i++) {
            if(coins < costs[i]) return res;
            coins -= costs[i];
            res ++;
        }
        return res;
    }

上面的方法没问题,但是不是最快的,因为排序要O(nlogn),
有没有办法在O(n)复杂度?

可以看到的是,有相同价格的冰棒,它们可以一起处理。
统计每个价格的冰棒的数量,
把这些数量一起处理掉,就可以避免遍历很长的数组。
先记下最大的价格,然后就在0~最大价格的范围内遍历即可。

    public int maxIceCream(int[] costs, int coins) {
        int n = costs.length;
        int maxCost = Arrays.stream(costs).max().getAsInt();
        int[] cnt = new int[maxCost+1];
        int res = 0;

        for(int cost : costs) {
            cnt[cost] ++;
        }

        for(int cost = 0; cost <= maxCost; cost++) {
            if(cnt[cost] == 0) continue;
            if(coins < cost) return res;
            int cur = Math.min(cnt[cost], coins/cost);
            coins -= cost * cur;
            res += cur;
        }
        return res;
    }

你可能感兴趣的:(leetcode,leetcode,算法)