→题目链接←
保存每个高度有多少个
如果目前剩的水的体积大于目前最矮的高度的数量,那么会使水面高度整体上升1,然后把目前的高度+1
然后再判断下以前是否已经有了+1后的高度,如果有把这两堆合并,再推进优先队列
直到最终高度后,只要初始高度小于等于最终高度,都加进ans
最后输出 最终高度 和 ans
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
struct node{
int num,val;
friend bool operator < (node a,node b){
return a.val>b.val;
}
};
int n,m,k;
priority_queue<node>q;
int a[404][404];
int h=0,ans=0;
int hh[10010];
int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1; i<=n; i++){
for(int j=1; j<=m; j++){
scanf("%d",&a[i][j]);
hh[a[i][j]]++;
}
}
for(int i=1; i<=10000; i++){
if(hh[i]){
node t;
t.num=hh[i];
t.val=i;
q.push(t);
}
}
while(k){
node t=q.top();
q.pop();
if(k>=t.num){
k-=t.num;
t.val++;
h=t.val;
if(hh[t.val]){
q.pop();
t.num+=hh[t.val];
q.push(t);
}
else q.push(t);
}
else break;
}
for(int i=1; i<=n; i++){
for(int j=1; j<=m; j++){
if(a[i][j]<=h)ans+=a[i][j];
}
}
printf("%d %d\n",h,ans);
return 0;
}