迷宮2(牛客網,最短路)

這一題主要就是思路加最短路。

首先你要知道這是一道求最短路的題目。
因爲起點或者終點是沒有辦法被圍起來的,所以只能從下邊或者左邊的一個點出發,到達上邊或者右邊的一個點(這個過程只能走上下左右四個方向,因爲蜥蜴可以走八個方向),過程的和最小即可,所以是多起點到多終點的最短路問題。

下面就是求最短路了,能找到就輸出,找不到就輸出-1。
當初求最短路一直超時,而且最短路的思路也還有點問題。
用spfa+優先隊列求最短路。

代碼:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<stack>
#define inf 0x3f3f3f3f3f3f3f3f
using namespace std;
typedef long long ll;
const int maxn=509;

int go[4][2]= {1,0,-1,0,0,1,0,-1};

bool vis[maxn][maxn];//記錄這個點是否在隊列中
ll dp[maxn][maxn];//記錄到達該點的最短路
ll mapp[maxn][maxn];
struct point//優先隊列
{
    int x,y;
    ll w;
    bool operator <(const point &a) const
    {
        return w>a.w;
    }
};

void init()
{
    memset(vis,0,sizeof(vis));
    memset(mapp,0,sizeof(mapp));
}

int main()
{
    int t,n,m;
    while(~scanf("%d%d%d",&t,&n,&m))
    {
        while(t--)
        {
            init();
            for(int i=1; i<=n; i++)
            {
                for(int j=1; j<=m; j++)
                {
                    dp[i][j]=inf;
                    scanf("%lld",&mapp[i][j]);
                    if(mapp[i][j]==0)//將路改爲-1;
                        mapp[i][j]=-1;
                    else if(mapp[i][j]==-1)//將牆改爲0;
                        mapp[i][j]=0;
                }
            }
            ll ans=-1;
            priority_queue<point>q;
            //從下邊或者左邊,到達上邊或者右邊
            for(int i=2; i<=n; i++)//左邊
            {
                if(mapp[i][1]!=-1)
                {
                    q.push(point{i,1,mapp[i][1]});
                    dp[i][1]=mapp[i][1];
                    vis[i][1]=1;
                }
            }
            for(int i=2; i<m; i++)//下邊
            {
                if(mapp[n][i]!=-1)
                {
                    q.push(point{n,i,mapp[n][i]});
                    dp[n][i]=mapp[n][i];
                    vis[n][i]=1;
                }
            }
            while(!q.empty())
            {
                point q1=q.top();
                q1.w=dp[q1.x][q1.y];
                if(q1.x==1||q1.y==m)//到達上邊或者右邊
                {
                    ans=q1.w;
                    break;
                }
                vis[q1.x][q1.y]=0;
                q.pop();
                for(int i=0; i<4; i++)
                {
                    int xx=q1.x+go[i][0],yy=q1.y+go[i][1];
                    ll ww=q1.w;
                    if(xx>0&&xx<=n&&yy>0&&yy<=m&&mapp[xx][yy]!=-1&&dp[xx][yy]>mapp[xx][yy]+ww)
                    {
                        dp[xx][yy]=mapp[xx][yy]+ww;
                        if(!vis[xx][yy])
                        {
                            q.push(point{xx,yy,dp[xx][yy]});
                            vis[xx][yy]=1;
                        }
                    }
                }
            }
            printf("%lld\n",ans);
        }
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章