感觉这题好水,我这种渣滓竟然想到了思路…
看到这道题,我先推了几个结论。
- 对于已固定高度的建筑群,其中的选取一定是选宽最小的。(因为高不影响结果)
- 将原数组按h值从小到大排序,h[k]是最后选取的建筑群的最小高度。
于是我们可以先将前k个push进去,从最小值开始枚举。
上马!
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
const int N = 1000002;
typedef long long ll;
priority_queue <ll> q;
ll ans, sum, res, tmp;
struct node {
ll x, y;
}s[N];
int n, k;
bool cmp(const node a, const node b) {
if(a.y > b.y || (a.y == b.y && a.x > b.x))
return 0;
return 1;
}
int main() {
scanf("%d %d", &n, &k);
for(int i = 1; i <= n; i ++) {
scanf("%lld %lld", &s[i].x, &s[i].y);
}
sort(s + 1, s + n + 1, cmp);
for(int i = 1; i <= k; i ++) {
q.push(s[i].x);
sum += s[i].x;
}
ans = sum * s[k].y;
for(int i = k + 1; i <= n; i ++) {
tmp = q.top();
res = s[i].y * (sum - tmp + s[i].x);
if(ans >= res) {
sum -= tmp;
sum += s[i].x;
ans = res;
q.pop();
q.push(s[i].x);
}
}
printf("%lld\n", ans);
return 0;
}
不要着急观看,偷偷告诉你,这份代码是错滴。
这是我的考场代码。发现问题没?我其实未将1结论理解透彻。由于后面的高度递增,我其实只要算前面宽的min即可,根本不需要什么判断!!!
上马!
#include<cstdio>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1000002;
typedef long long ll;
priority_queue <ll> q;
ll ans, sum, res, tmp;
struct node {
ll x, y;
}s[N];
int n, k;
bool cmp(const node a, const node b) {
if(a.y > b.y || (a.y == b.y && a.x > b.x))
return 0;
return 1;
}
int main() {
scanf("%d %d", &n, &k);
for(int i = 1; i <= n; i ++) {
scanf("%lld %lld", &s[i].x, &s[i].y);
}
sort(s + 1, s + n + 1, cmp);
for(int i = 1; i <= k; i ++) {
q.push(s[i].x);
sum += s[i].x;
}
ans = sum * s[k].y;
for(int i = k + 1; i <= n; i ++) {
tmp = q.top();
res = s[i].y * (sum - tmp + s[i].x);
sum -= tmp;
sum += s[i].x;
ans = min(ans, res);
q.pop();
q.push(s[i].x);
}
printf("%lld\n", ans);
return 0;
}
至于为什么先pop再push我有话说:其实可以先push再pop。只是就算s[i].x大于q.top(),最好的情况早已算过了,也不会错。
如有错误,请大佬在评论区指出。谢谢!(台词说了好多遍了emmmm )