【NOIP2017】【Luogu P3956】【SPFA】棋盘

Luogu P3956

本题是一道简单的SPFA

具体看程序

#include<iostream>
#include<cstdio>
using namespace std;
const int wayx[4]={1,0,-1,0},wayy[4]={0,1,0,-1};//4个方向
int n,m,cost[200][200],a[200][200],magic[200][200],x,y,c;
bool visit[200][200];//标志某个点有没有在队列里面
struct data
{
    int x,y;
}que[100000];//队列内的点的坐标
void first()
{
    for (int i=1;i<=m;i++)
        for (int j=1;j<=m;j++)
            cost[i][j]=2147483640;
    cost[1][1]=0;
    //初始化，起点价值为0，其他均为无穷大
}
void spfa()//程序主体 SPFA
{
    int head=0,tail=1;
    que[head].x=1;que[head].y=1;
    visit[1][1]=true;
    while (head!=tail)
    {
        int nowx=que[head].x,nowy=que[head].y;
        if (magic[nowx][nowy]) a[nowx][nowy]=magic[nowx][nowy];
        //如果这个点用过魔法，那么让这个点标记为变的颜色
        for (int i=0;i<4;i++)
        {
            int movex=nowx+wayx[i],movey=nowy+wayy[i],nowc=0;
            if (movex<0||movex>m) continue;
            if (movey<0||movey>m) continue;
            //防止越界
            if (a[movex][movey]==1&&a[nowx][nowy]==2) nowc+=1;
            if (a[movex][movey]==2&&a[nowx][nowy]==1) nowc+=1;
            if (a[movex][movey]==0)
            { 
                if (magic[nowx][nowy]) nowc=cost[movex][movey]-cost[nowx][nowy]+1;
                else
                {
                    nowc+=2;
                }
            }
            if (cost[movex][movey]>cost[nowx][nowy]+nowc)//松弛
            {
                if (!visit[movex][movey])
                {
                    que[tail].x=movex;
                    que[tail].y=movey;
                    tail++;
                    visit[movex][movey]=true;
                }
                cost[movex][movey]=cost[nowx][nowy]+nowc;
                if (nowc==2) magic[movex][movey]=a[nowx][nowy];
            }
        }
        visit[nowx][nowy]=false;
        if (magic[nowx][nowy]) a[nowx][nowy]=0;
        magic[nowx][nowy]=0;
        head++;
    }
}
int main()
{
    cin>>m>>n;
    for (int i=1;i<=n;i++)
    {
        cin>>x>>y>>c;
        a[x][y]=c+1;
    }
    first();
    spfa();
    if (cost[m][m]==2147483640) cout<<-1;//即没有路径可以到达终点
    else cout<<cost[m][m];
    return 0;
}