2024华为OD试题及答案-A038-统一限载货物数最小值

题目描述

火车站附近的货物中转站负责将到站货物运往仓库，小明在中转站负责调度2K辆中转车（K辆干货中转车，K辆湿货中转车）。

货物由不同供货商从各地发来，各地的货物是依次进站，然后小明按照卸货顺序依次装货到中转车，一个供货商的货只能装到一辆车上，不能拆装，但是一辆车可以装多家供货商的货；

中转车的限载货物量由小明统一指定，在完成货物中转的前提下，请问中转车的统一限载货物数最小值为多少。

输入描述

• 第一行 length 表示供货商数量 1 <= length <= 10^4
• 第二行 goods 表示供货数数组 1 <= goods[i] <= 10^4
• 第三行 types表示对应货物类型，types[i]等于0或者1，其中0代表干货，1代表湿货
• 第四行 k表示单类中转车数量 1 <= k <= goods.length

输出描述

运行结果输出一个整数，表示中转车统一限载货物数

备注

中转车最多跑一趟仓库

用例

输入	4 3 2 6 3 0 1 1 0 2
输出	6
说明	货物1和货物4为干货，由2辆干货中转车中转，每辆车运输一个货物，限载为3 货物2和货物3为湿货，由2辆湿货中转车中转，每辆车运输一个货物，限载为6 这样中转车统一限载货物数可以设置为6（干货车和湿货车限载最大值），是最小的取值

输入	4 3 2 6 8 0 1 1 1 1
输出	16
说明	货物1为干货，由1辆干货中转车中转，限载为3 货物2、货物3、货物4为湿货，由1辆湿货中转车中转，限载为16 这样中转车统一限载货物数可以设置为16（干货车和湿货车限载最大值），是最小的取值

解题思路分析

我们需要设计一个算法来计算最小货物转运次数。货物分为干货和湿货，每辆货车只能装一种货物，并且每辆车只能装一个供应商的货物。假设有两个类别的货车（K辆干货车和K辆湿货车），我们需要在满足条件的情况下找到中转车的统一限载货物数的最小值。

问题分解：

1. 输入：

   • 第一行：供应商数量length。
   • 第二行：每个供应商提供的货物数量数组goods。
   • 第三行：货物类型数组types（0代表干货，1代表湿货）。
   • 第四行：每类中转车的数量k。

2. 输出：

   • 一个整数，表示中转车统一限载货物数的最小值。

步骤：

1. 二分查找：利用二分查找方法来确定最小的限载货物数。我们从1到所有货物总量的最大值中间进行二分查找。
2. 验证函数：在每次二分查找中验证当前的限载货物数是否可行。利用贪心算法验证是否能在给定的中转车数量内完成转运。
3. 分类计数：分别计算干货和湿货的货物量并分别用对应的中转车来运输。

C++程序实现

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

bool canTransport(const vector<int>& goods, const vector<int>& types, int k, int maxLoad) {
    int dryCars = k, wetCars = k;
    for (int i = 0; i < goods.size(); ++i) {
        int numCarsNeeded = (goods[i] + maxLoad - 1) / maxLoad; // 向上取整计算所需车数
        if (types[i] == 0) {
            dryCars -= numCarsNeeded;
        } else {
            wetCars -= numCarsNeeded;
        }
        if (dryCars < 0 || wetCars < 0) return false;
    }
    return true;
}

int findMinimumMaxLoad(int length, const vector<int>& goods, const vector<int>& types, int k) {
    int left = 1, right = *max_element(goods.begin(), goods.end()), result = right;
    while (left <= right) {
        int mid = left + (right - left) / 2;
        if (canTransport(goods, types, k, mid)) {
            result = mid;
            right = mid - 1;
        } else {
            left = mid + 1;
        }
    }
    return result;
}

int main() {
    int length, k;
    cin >> length;
    vector<int> goods(length), types(length);
    for (int i = 0; i < length; ++i) cin >> goods[i];
    for (int i = 0; i < length; ++i) cin >> types[i];
    cin >> k;

    int result = findMinimumMaxLoad(length, goods, types, k);
    cout << result << endl;

    return 0;
}

Python程序实现

def can_transport(goods, types, k, max_load):
    dry_cars = k
    wet_cars = k
    for i in range(len(goods)):
        num_cars_needed = (goods[i] + max_load - 1) // max_load  # 向上取整计算所需车数
        if types[i] == 0:
            dry_cars -= num_cars_needed
        else:
            wet_cars -= num_cars_needed
        if dry_cars < 0 or wet_cars < 0:
            return False
    return True

def find_minimum_max_load(length, goods, types, k):
    left, right = 1, max(goods)
    result = right
    while left <= right:
        mid = (left + right) // 2
        if can_transport(goods, types, k, mid):
            result = mid
            right = mid - 1
        else:
            left = mid + 1
    return result

# 输入
length = int(input().strip())
goods = list(map(int, input().strip().split()))
types = list(map(int, input().strip().split()))
k = int(input().strip())

# 计算并输出结果
result = find_minimum_max_load(length, goods, types, k)
print(result)

解释

上述代码使用了二分查找法来确定中转车的最小限载货物数。每次二分查找中，我们通过验证函数来判断当前的限载货物数是否可行。验证函数通过模拟每种货物类型的中转车数量来确保满足条件。C++和Python的实现逻辑一致，只是语法有所不同。