题目描述
现有司机N * 2人,调度中心会将所有司机平分给A、B两个区域
- 第 i 个司机去A可得收入为income[i][0],
- 第 i 个司机去B可得收入为income[i][1],
返回所有调度方案中能使所有司机总收入最高的方案,是多少钱
题目解析
暴力递归
从左到右尝试模型,每个司机都有两种方案:
- 分到A区域
- 分到B区域
但是有一共限制条件,A区域分而且必须只能分的司机N个
求总收入最高的方案
尝试一:
int process(std::vector<std::vector<int>> incomes, int idx, int restA, int restB, std::map<std::string, int> &cache){
if(idx == incomes.size()){
return 0;
}
std::string key = std::to_string(idx) + "_" + std::to_string(restA) + "_" + std::to_string(restB);
if(cache.count(key)){
return cache[key];
}
if(restA == 0){
return cache[key] = incomes[idx][1] + process(incomes, idx + 1, restA, restB - 1, cache);
}else if(restB == 0){
return cache[key] = incomes[idx][0] + process(incomes, idx + 1, restA - 1, restB, cache);
}
int p1 = incomes[idx][0] + process(incomes, idx + 1, restA - 1, restB, cache);
int p2 = incomes[idx][1] + process(incomes, idx + 1, restA, restB - 1, cache);
return cache[key] = std::max(p1, p2);
}
public:
int maxMoney(std::vector<std::vector<int>> incomes){
int N = incomes.size();
if(N <= 1 && N & 1 == 1){
return -1;
}
std::map<std::string, int> cache;
return process(incomes, 0, N >> 1, N >> 1, cache);
}
尝试2(可以减少一个参数):
int process1(std::vector<std::vector<int>> incomes, int idx, int restA, std::map<std::string, int> &cache){
if(idx == incomes.size()){
return 0;
}
std::string key = std::to_string(idx) + "_" + std::to_string(restA);
if(cache.count(key)){
return cache[key];
}
if(incomes.size() - idx == restA){
return cache[key] = incomes[idx][0] + process1(incomes, idx + 1, restA - 1, cache);
}
if(restA == 0){
return cache[key] = incomes[idx][1] + process1(incomes, idx + 1, restA, cache);
}
int p1 = incomes[idx][0] + process1(incomes, idx + 1, restA - 1, cache);
int p2 = incomes[idx][1] + process1(incomes, idx + 1, restA, cache);
return cache[key] = std::max(p1, p2);
}
int maxMoney1(std::vector<std::vector<int>> incomes){
int N = incomes.size();
if(N <= 1 && N & 1 == 1){
return -1;
}
std::map<std::string, int> cache1;
return process1(incomes, 0, N >> 1, cache1);
}
思路
思路:
- 先让所有司机去B市,然后再从其中找出N名司机去A市
- 去A市的司机,收入必定要比在B市更赚。其赚得最多的N名就是我们要找的N名司机。
- 可以通过用“A市的收入 - B市的收入”去最大的N名即可实现收入最大化。
- 其中“A市的收入 - B市的收入”这个值可正可负,负数表示该司机去A市亏了。
- 所以这道题,仅仅需要通过排序“A市的收入 - B市的收入”即可正确解答。