牛客OI周赛8-提高组 A 用水填坑

最新推荐文章于 2020-05-27 09:54:04 发布

欣君

最新推荐文章于 2020-05-27 09:54:04 发布

阅读量215

点赞数

分类专栏：【搜索】 ----bfs 【贪心】

本文链接：https://blog.csdn.net/xin_jun/article/details/89384965

版权

【贪心】同时被 3 个专栏收录

12 篇文章 0 订阅

订阅专栏

----bfs

5 篇文章 0 订阅

订阅专栏

【搜索】

3 篇文章 0 订阅

订阅专栏

95%的分很好拿，只要对每一个高度都跑一遍dfs/bfs就行了，但最后5%的分有难度。

首先我们可以知道，最外面一圈的高度和蓄水量，是固定的。我们可以考虑从外圈向内圈进行反向搜索操作。

用一个堆/优先队列作为搜索队列，维护已确定蓄水量的单元格中，高度+蓄水量 h'[x][y] 最小的单元格。

每次取出堆顶元素，其周围的单元格高度为 h[i][j] 。

如果 h[i][j]<h'[x][y] ，必定可以将这个单元格填补到高度，总蓄水量增加 h'[x][y]-h[x'][y'] ， h'[i][j]=h'[x][y] 。

如果 $h[i][j]\geq h'[x][y]$ ，那说明这个单元格无法蓄水， h'[i][j]=h[i][j] 。

然后将新确定高度+蓄水量 h'[i][j] 的单元格，压入堆中。

#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);
}