Tram

題目：

給出站點以及起點和終點，在每個站點，給出該站點可以到達地的站點個數（可以認爲兩個站點之間可以到達那麼距離爲1），求出起點與終點最短距離；

代碼：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
#define inf 9999999
int  mapp[111][111];
int book[111],dis[111];

int main()
{
    int i,j,k,n,a,b;;
    scanf("%d %d %d",&n,&a,&b);
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            if(i==j)
                mapp[i][j]=0;
            else
                mapp[i][j]=inf;
        }
    }
    for(i=1;i<=n;i++)
    {
        int k,h;
        scanf("%d",&k);
        for(j=1;j<=k;j++)
        {
            scanf("%d",&h);
            if(j==1)
                mapp[i][h]=0;
            else
                mapp[i][h]=1;
        }
    }
    memset(book,0,sizeof(book));
    for(i=1;i<=n;i++)
        dis[i]=mapp[a][i];
    dis[a]=0;
    book[a]=1;
    for(i=1;i<=n;i++)
    {
        int minn=inf,u;
        for(j=1;j<=n;j++)
        {
            if(!book[j]&&minn>dis[j])
            {
                minn=dis[j];
                u=j;
            }
        }
        book[u]=1;
        for(k=1;k<=n;k++)
            dis[k]=min(dis[k],dis[u]+mapp[u][k]);
    }
    if(dis[b]==inf)
        printf("-1\n");
    else
        printf("%d\n",dis[b]);
    return 0;
}