算法竞赛入门经典(训练指南)第28页,例题12
注意二分的时候M = (L+R+1)/2,二分法为了避免死循环,根据情况M=(L+R)/2 || M=(L+R+1)/2
题解对输入的多个字符串编号采用了map<>的写法以后可以借鉴下
题解就是假设最小的品质因子是x,则删除所有品质因子小于x的配件,然后判断剩下配件能否组装出一个总价小于b的电脑,能的话,说明答案ans>=x,否则ans<x
至于判断能否组装出总价小于b的电脑,直接在每种配件中选择价格最便宜的即可。
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <map>
#include <vector>
#define MAXN 1010
using namespace std;
int cnt; // 配件类型数
int n,b;
// 以后处理输入的很多字符串,需要对其哈希承1-n的数字可以用此方法
map<string, int> m; // 配件类型及其对应的类型号
struct Component {
int price;
int quality;
};
vector<Component> comp[MAXN];
// 对输入的字符串哈希成编号
int ID(string s) {
if(m.count(s) == 0) {
m[s] = cnt++;
}
return m[s];
}
// q代表选中配件的最小品质
bool isok(int q) {
int totalPrice = 0; // 所需的总价格
for(int i = 0;i < cnt;i++) {
int temp = 0; // 统计多少个配件品质>=q
int minPrice = 1e8;
for(int j = 0;j <= comp[i].size();j++) {
if(comp[i][j].quality >= q) {
temp++;
minPrice = min(minPrice,comp[i][j].price);
}
}
// 该类配件缺少
if(temp == 0) return false;
totalPrice += minPrice;
if(totalPrice > b) return false;
}
return true;
}
int main() {
int T;
scanf("%d",&T);
while(T--) {
cnt = 0;
n = b = 0;
m.clear();
for(int i = 0;i < MAXN;i++) comp[i].clear();
scanf("%d%d",&n,&b);
int maxq = -1; // 最大品质
for(int i = 0;i < n;i++) {
char kind[30],name[30];
int p,q;
scanf("%s%s%d%d",kind,name,&p,&q);
maxq = max(maxq,q);
comp[ID(kind)].push_back((Component){p,q});
}
int L = 0,R = maxq;
while(L < R) {
int M = (L+R+1)/2;
if(isok(M)) {
L = M;
} else {
R = M - 1;
}
}
printf("%d\n",L);
}
return 0;
}