牛客网 Wannafly交流赛1 D 迷宫2 （思维+bfs）

题目链接

解析:
其实题意就是要你造一幢墙来堵住两边，那么我们就可以用墙来作为处理的对象
首先如果要堵住的话，墙的两边肯定是要碰边界的，假如说墙一开始是从第一列的某一处开始的，那么最后终点一定是第一行的某一处或者是第m列的某一处，这样才能把他们
分隔出来，这样从第n行出发的也是同理。所以我们就只要从第1列或第n行开始进行bfs，状态量是到当前节点最小的花费

#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;

const int MAXN = 500+10;
#define INF 0x3f3f3f3f3f3f3f3f
typedef struct node
{
int x;
int y;
lli d;

friend bool operator < (node a,node b)
{
return a.d>b.d;
}
}node;

lli mp[MAXN][MAXN];
priority_queue<node> que;

int dir[4][2]={-1,0,0,1,1,0,0,-1};
lli dis[MAXN][MAXN];
int main()
{
int q;
int n,m;
scanf("%d%d%d",&q,&n,&m);
while(q--)
{
while(que.size()) que.pop();
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
scanf("%lld",&mp[i][j]);
if(mp[i][j]==-1) mp[i][j]=0;
else if(mp[i][j]==0) mp[i][j]=-1;
dis[i][j]=INF;
}
}

for(int i=1;i<=n;i++)
{
if(mp[i][1]==-1) continue;
que.push(node{i,1,mp[i][1]});
dis[i][1]=mp[i][1];
}

for(int i=1;i<=m;i++)
{
if(mp
[i]==-1) continue;
que.push(node{n,i,mp
[i]});
dis
[i]=mp
[i];
}
lli ans=-1;
while(que.size())
{

node tmp=que.top();
que.pop();
if(tmp.x==1||tmp.y==m)
{
ans=tmp.d;
break;
}

for(int i=0;i<4;i++)
{
int nx=tmp.x+dir[i][0];
int ny=tmp.y+dir[i][1];

if(nx>=1&&nx<=n&&ny>=1&&ny<=m&&mp[nx][ny]+tmp.d<dis[nx][ny]&&mp[nx][ny]!=-1)
{
que.push(node{nx,ny,mp[nx][ny]+tmp.d});
dis[nx][ny]=mp[nx][ny]+tmp.d;
}
}

}
printf("%lld\n",ans);

}
return 0;
}