用水填坑 (牛客)优先队列

链接: https://ac.nowcoder.com/acm/contest/403/A
题意:
现有一块n * m的地,每块1*1的地的高度为 h i, j,在n * m的土地之外的土地高度均为0(即你可以认为最外圈无法积水)
现在下了一场足够大的雨,试求出这块地总共积了多少单位体积的水
思路:
先说说我的错误思路,一开始的我思路是枚举每个点的四周是否都比自身,是就提升到自身四周最低的那个高度,这个思路错误在……还是看样例说话吧

3 3 3 3
3 0 0 3
3 0 0 3
3 3 3 3
按照我的思路这道题为积水为0,所以hack

新思路:
我们可以把四周的点全部存进优先队列,并标记,每次把高度最小的出列,每个点走一次,然后判断这个点四周有没有比它更矮的点,有就积水到一样的高度,然后标记进队,因为每次都是高度最小的优先,所以维护积水不会溢出

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e3+5;
int mp[N][N];
bool vis[N][N];
int dir[4][2]={1,0,0,1,-1,0,0,-1};
struct node
{
	int x,y,high;
	bool operator < (const node &a)const
	{
		return high>a.high;
	}
};
int main()
{
	int n,m;
	cin>>n>>m;
	priority_queue<node>que;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			cin>>mp[i][j];
			if(i==n||j==m||i==1||j==1)
			{
			   vis[i][j]=1;
			   que.push(node{i,j,mp[i][j]});
			}
		}
	}
	ll ans=0;
	while(que.size())
	{
		node tmp=que.top();
		que.pop();
		for(int i=0;i<4;i++)
		{
		   int tx=tmp.x+dir[i][0],ty=tmp.y+dir[i][1];
		   if(tx>0&&tx<=n&&ty>0&&ty<=m&&!vis[tx][ty])
		   {
		   	vis[tx][ty]=1;
		   	if(mp[tx][ty]<tmp.high)
			{
			   ans+=tmp.high-mp[tx][ty];
			   mp[tx][ty]=tmp.high;
			}
			que.push(node{tx,ty,mp[tx][ty]});
		   }
		}
	}
	/*
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			cout<<mp[i][j]<<' ';
		}
		cout<<endl;
	}
	*/
	cout<<ans<<endl;
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值