题目链接:http://acm.nyist.net/JudgeOnline/problem.php?pid=1238
题目是中文题,很好理解,要注意的是公交车都是单向去走的,所以第二个样例才会是NO。
思路:由于数据量非常小,写个N^2的dijkstra就能跑过去了。我们可以把每条公交线路放进一个vector里面存,并且把每个公交站点所在的线路也放进一个vector里面存起来。每次跑最短路,更新站点所在公交线路的后面的点的距离就好了。
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
vector <int> road[505], in[505];
int n, dis[505];
bool vst[505], vst_road[505];
void dijkstra()
{
int pos, Min, key, k;
memset(vst, 0, sizeof(vst));
for(int i = 1; i <= n; i++)
dis[i] = INF;
dis[1] = -1;
for(int j = 0; j < n; j++)
{
Min = INF;
for(int i = 1; i <= n; i++)
{
if(!vst[i] && dis[i] < Min)
{
pos = i;
Min = dis[i];
}
}
if(Min == INF)
break;
vst[pos] = 1;
for(int i = 0; i < in[pos].size(); i++)
{
key = in[pos][i];
k = 0;
while(road[key][k++] != pos);
for(; k < road[key].size(); k++)
{
int tmp = road[key][k];
if(!vst[tmp] && dis[tmp] > dis[pos] + 1)
dis[tmp] = dis[pos] + 1;
}
}
if(dis[n] != INF)
break;
}
dis[1] = 0;
}
int main()
{
int T, m, x;
char ch;
cin >> T;
while(T--)
{
cin >> m >> n;
for(int i = 0; i < 505; i++)
{
in[i].clear();
road[i].clear();
}
for(int i = 0; i < m; i++)
{
while(1)
{
scanf("%d", &x);
in[x].push_back(i);
road[i].push_back(x);
ch = getchar();
if(ch == '\n')
break;
}
}
dijkstra();
if(dis[n] == INF)
cout << "NO" << endl;
else
cout << dis[n] << endl;
}
}