bzoj 3358: [Usaco2004 Jan]洪水

→题目链接←

保存每个高度有多少个

如果目前剩的水的体积大于目前最矮的高度的数量，那么会使水面高度整体上升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;
}