阿里算法笔试模拟题精解之“Tom 爱吃巧克力”

查看题目：Tom 爱吃巧克力

Tom 非常喜欢巧克力，他上次买的巧克力吃完了，所以他打算再去买 k 块巧克

力回来 (1<=k<=1e5)，他又是一个非常节俭的一个人，所以他想花最少的钱去买巧

克力。

现在有 n 家卖巧克力的店 (1<=n<=1e5)，每个店的巧克力都限购 bi 块 ( 最多只

能买 bi 块 ,1<=bi<=1e5)，每块的价格是 ai(1<=ai<=1e9)，请问 Tom 买 k 块巧克

力最少要花多少钱？题目保证 n 个 bi 的总和大于等于 k。

输 入 卖 巧 克 力 的 店 的 个 数 n(1<=n<=1e5)； 打 算 去 买 的 巧 克 力 块 数

k(1<=k<=1e5)；和一个数组 m, 其中 mi = ai, bi 表示第 i 家巧克力店的巧克力的价格

和限购块数

输出一个数，表示 Tom 买 k 块巧克力花的最少钱数

示例 1

输入：

2

5

[[4,5],[2,4]]

输出：

12

解题思路：贪心

根据题意，可以得知这道题可以运用贪心算法，策略是每次都去买最便宜的巧

克力。

由于题目给的二维数组是乱序的，可以根据巧克力的价格对二维数组从小到大进

行排序，便于 Tom 选择最便宜的巧克力。Arrays 类中只提供了一维数组的排序，如

果要用 Arrays 对二维数组排序，需要重写 Comparator 里的 compare 方法。

排序完成后，接下来操作就比较简单了。遍历数组，优先买便宜的巧克力，直到

达到限购块数，再去下一家巧克力店。最终买到 k 块巧克力时 Tom 花的钱最少。

给出了思路没有给代码，代码不一定相同单思路差不多

public static int leastMoney() {
        int n = 3;//店铺数
        int k = 8;//打算买的块数
        int total = 0;//总价
        int[][] arr = {{4, 5}, {2, 4},{3,3}};
        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0] - o2[0];
            }
        });
        for (int i = 0; i < n; i++) {
            if (k>arr[i][1]){
                total += arr[i][0] * arr[i][1];
                k = k - arr[i][1];
            }else if ( k<arr[i][1] && k != 0){
                total += arr[i][0] * k;
            }
        }
        return total;
    }