题目大意:给T组数据,之后一行n,m,k分别表示n个建筑,初始m个兵,一个兵生产一个兵要k时间。接下来n个数分别表示建造对应建筑要消耗的时间,求最少要多少时间?一个兵生产后还可以工作,工作后要将其移除。
思路:贪心,类似于哈夫曼建树的过程,具体思路见模拟哈夫曼建树(好题)
Code:
错误思路
#include <iostream>
#include <queue>
#include <algorithm>
#include <stdio.h>
using namespace std;
const int maxn = (int)1e5+5;
typedef long long LL;
int num[maxn];
int n,m,k;
LL mT, now;
bool cmp (int a, int b) {
return a > b;
}
int main() {
int T;
scanf("%d", &T);
while (T--) {
scanf("%d%d%d", &n, &m, &k);
for (int i = 0; i < n; i++) {
scanf("%d", &num[i]);
}
sort(num, num + n, cmp);
mT = now = 0; //目前最大时间
for (int i = 0; i < n;) {
//只有1只并且不止有1栋建筑
if (m == 1 && i < n - 1) {
now += k;
m *= 2;
continue;
}
//只有1只并且只有1栋建筑
if (m == 1 && i == n - 1) {
mT = max(mT, now + num[i]);
i++;
continue;
}
//目前已经够人 全拉过去施工
if (m + i >= n) {
mT = max(mT, now + num[i]);
break;
}
//如果本次建筑会让全局时间得到更新说明要也早建越好
if (num[i] + now >= mT) {
mT = max(mT, now + num[i]);
m--;
i++;
continue;
}
//
now += k;
m *= 2;
}
printf("%lld\n", mT);
}
return 0;
}
正解
#include <iostream>
#include <stdio.h>
#include <queue>
#include <vector>
using namespace std;
int n,m,k;
priority_queue<int, vector<int>, greater<int> > num;
int main() {
int T, tmp;
scanf("%d", &T);
while (T--) {
while (!num.empty()) {
num.pop();
}
scanf("%d%d%d", &n, &m, &k);
for (int i = 0; i < n; i++) {
scanf("%d", &tmp);
num.push(tmp);
}
while (n > m) {
n--;
num.pop();
tmp = num.top() + k; //最小的那个用来生产新胚胎
num.pop();
num.push(tmp);
}
while (num.size() > 1) {
num.pop();
}
printf("%d\n",num.top());
}
return 0;
}