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;
}