牛客OI周賽8-提高組 A 用水填坑

95%的分很好拿，只要對每一個高度都跑一遍dfs/bfs就行了，但最後5%的分有難度。

首先我們可以知道，最外面一圈的高度和蓄水量，是固定的。我們可以考慮從外圈向內圈進行反向搜索操作。

用一個堆/優先隊列作爲搜索隊列，維護已確定蓄水量的單元格中，高度+蓄水量最小的單元格。

每次取出堆頂元素，其周圍的單元格高度爲。

如果，必定可以將這個單元格填補到高度，總蓄水量增加，。

如果，那說明這個單元格無法蓄水，。

然後將新確定高度+蓄水量的單元格，壓入堆中。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=1010;
int g[2][4]={{-1,0,0,1},{0,-1,1,0}};
int a[MAXN][MAXN],vis[MAXN][MAXN];

struct Node
{
	int x,y,h;
	Node(){	}
	Node(int _x,int _y,int _h)
	{
		x=_x;
		y=_y;
		h=_h;
	}
	bool friend operator<(Node n1,Node n2)
	{
		return n1.h>n2.h;
	}
};

priority_queue<Node> q;

int main()
{
	int n,m,i,j;
	Node now,nxt;
	long long ans;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			scanf("%d",&a[i][j]);
	for(i=1;i<=n;i++)
	{
		q.push(Node(i,1,a[i][1]));
		q.push(Node(i,m,a[i][m]));
		vis[i][1]=1;
		vis[i][m]=1;
	}
	for(j=2;j<=m-1;j++)
	{
		q.push(Node(1,j,a[1][j]));
		q.push(Node(n,j,a[n][j]));
		vis[1][j]=1;
		vis[n][j]=1;
	}
	ans=0;
	while(!q.empty())
	{
		now=q.top();
		q.pop();
		for(i=0;i<4;i++)
		{
			nxt=Node(now.x+g[0][i],now.y+g[1][i],a[now.x+g[0][i]][now.y+g[1][i]]);
			if(nxt.x<1||nxt.x>n||nxt.y<1||nxt.y>m||vis[nxt.x][nxt.y])
				continue;
			if(nxt.h<now.h)
			{
				ans+=now.h-nxt.h;
				nxt.h=now.h;
			}
			q.push(nxt);
			vis[nxt.x][nxt.y]=1;
		}
	}
	printf("%lld\n",ans);
}