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

题目链接


解析:

其实题意就是要你造一幢墙来堵住两边,那么我们就可以用墙来作为处理的对象

首先如果要堵住的话,墙的两边肯定是要碰边界的,假如说墙一开始是从第一列的某一处开始的,那么最后终点一定是第一行的某一处或者是第m列的某一处,这样才能把他们

分隔出来,这样从第n行出发的也是同理。所以我们就只要从第1列或第n行开始进行bfs,状态量是到当前节点最小的花费


#include 
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 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[n][i]==-1) continue;
            que.push(node{n,i,mp[n][i]});
            dis[n][i]=mp[n][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

你可能感兴趣的:(思维题,BFS,STL容器)