第八届河南省赛 最少换乘（最短路）

题目链接：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
!= 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
== INF)
cout << "NO" << endl;
else
cout << dis
<< endl;
}
}