洛谷 P1346 电车(最短路_Floyd)

传送门


每个点向它所联通的点建单向边,开关默认指向的边权为0,其他为1,然后跑最短路(FLoyd,SPFA,Dijkstra均可),由于2<=N<=100,我们用邻接矩阵跑FLoyd就可以了。

Code:

#include
#include
#include

const int INF=1e9;

int a[110][110];
int n,st,ed;

int main()
{
    scanf("%d %d %d",&n,&st,&ed);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
            a[i][j]=INF;
        a[i][i]=0;
    }
    for(int i=1;i<=n;i++)
    {
        int k;scanf("%d",&k);
        for(int j=1;j<=k;j++)
        {
            int x;scanf("%d",&x);
            if(j==1) a[i][x]=0;
            else a[i][x]=1;
        }
    }
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            if(i!=k) for(int j=1;j<=n;j++)
                if(i!=j && k!=j && a[i][k]+a[k][j]if(a[st][ed]==INF) printf("-1");
    else printf("%d",a[st][ed]);
}

你可能感兴趣的:(最短路)