卡车上的最大单元数

题目

请你将一些箱子装在 一辆卡车 上。给你一个二维数组 boxTypes ,其中 boxTypes[i] = [numberOfBoxesi, numberOfUnitsPerBoxi] :

numberOfBoxesi 是类型 i 的箱子的数量。
numberOfUnitsPerBoxi 是类型 i 每个箱子可以装载的单元数量。
整数 truckSize 表示卡车上可以装载 箱子 的 最大数量 。只要箱子数量不超过 truckSize ,你就可以选择任意箱子装到卡车上。

返回卡车可以装载 单元 的 最大 总数。

示例 1:

输入:boxTypes = [[1,3],[2,2],[3,1]], truckSize = 4
输出:8
解释:箱子的情况如下:

  • 1 个第一类的箱子,里面含 3 个单元。
  • 2 个第二类的箱子,每个里面含 2 个单元。
  • 3 个第三类的箱子,每个里面含 1 个单元。
    可以选择第一类和第二类的所有箱子,以及第三类的一个箱子。
    单元总数 = (1 * 3) + (2 * 2) + (1 * 1) = 8
    示例 2:

输入:boxTypes = [[5,10],[2,5],[4,7],[3,9]], truckSize = 10
输出:91

提示:

1 <= boxTypes.length <= 1000
1 <= numberOfBoxesi, numberOfUnitsPerBoxi <= 1000
1 <= truckSize <= 106

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-units-on-a-truck
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一:贪心

  • 解题思路
    找出单元最大的箱子,先装最大的箱子,然后再装其次大的。
public static int maximumUnits(int[][] boxTypes, int truckSize) {
        int sum = 0;
        while(truckSize>0)
        {
            int maxValue=0,maxIndex=0;
            for (int i = 0; i < boxTypes.length; i++) {
                if(boxTypes[i][0]>0)
                {
                    if(boxTypes[i][1]>maxValue)
                    {
                        maxValue=boxTypes[i][1];
                        maxIndex=i;
                    }
                }
            }
            if(maxIndex==-1)
            {
                break;
            }
            if(truckSize>=boxTypes[maxIndex][0])
            {
                sum+=boxTypes[maxIndex][0]*boxTypes[maxIndex][1];
                truckSize=truckSize-boxTypes[maxIndex][0];
                boxTypes[maxIndex][0]=0;
            }
            else
            {
                sum+=truckSize*boxTypes[maxIndex][1];
                truckSize=0;
            }
        }
        return sum;
    }

解法二:排序+贪心

  • 在上一解法的思路基础上,先排序,再遍历
public static int maximumUnits2(int[][] boxTypes, int truckSize)
    {
        Arrays.sort(boxTypes, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o2[1]-o1[1];
            }
        });
        int sum = 0;
        for (int i = 0; i < boxTypes.length; i++) {
            if(truckSize>0)
            {
                if(truckSize>=boxTypes[i][0])
                {
                    sum+=boxTypes[i][0]*boxTypes[i][1];
                    truckSize=truckSize-boxTypes[i][0];
                }
                else {
                    sum+=truckSize*boxTypes[i][1];
                    truckSize=0;
                }
            }
            else {
                break;
            }
        }
        return sum;
    }
posted @ 2022-11-15 09:47  花茶冰糖  阅读(38)  评论(0编辑  收藏  举报