POJ 1847 Tram
原题链接:vjudge传送门
题目大意:
给定n个路口,每个路口通向多条路,默认通向第一条路,若想通向该路口的任何一条路,需要拨动一次开关,问给定路口A,B,从A出发到B最少需要拨动多少次开关。
例:
3 2 1 三个路口,从2到1号路口最少要拨动多少次开关
2 2 3 1号路口,通向两条路,默认通向2号路口的路,还有通向3号路口的路(需拨动一次开关)
2 3 1 2号路口,如上类比
2 1 2 3号路口
具体思路:
通向默认路口的路径权值为0,通向其它路口的路径权值为1,然后求解最短路即可。
具体代码:
#include<iostream>
#include<cmath>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
int const N = 110, M = 1e4+5;
int const INF = 0x3f3f3f3f;
int h[N],d[N],visit[N];
int cnt = 0; //用于前向星建图
int n,start,des;
struct Node {
int v;
int w;
int next;
}e[M];
void add(int u, int v, int w)
{
e[cnt].v = v;
e[cnt].w = w;
e[cnt].next = h[u];
h[u] = cnt++;
}
int spfa()
{
memset(visit, 0, sizeof(visit));
memset(d, 0x3f, sizeof(d));
queue<int> q;
q.push(start);
d[start] = 0;
while (q.size())
{
int t = q.front();
q.pop();
visit[t] = 0;
for (int i = h[t]; ~i; i = e[i].next)
{
int v = e[i].v;
int w = d[t] + e[i].w;
if (d[v] > w ) {
d[v] = w;
if (!visit[v]) {
q.push(v);
visit[v] = 1;
}
}
}
}
return d[des];
}
int main()
{
memset(h, -1, sizeof(h));
scanf("%d%d%d", &n, &start, &des);
for (int i = 1; i <= n; i++)
{
int num;
scanf("%d", &num);
int cnt = 1;
while (num--)
{
int v;
scanf("%d", &v);
if (cnt == 1)
add(i, v, 0);
else add(i, v, 1);
cnt++;
}
}
int ans = spfa();
if (ans != INF)printf("%d\n",ans );
else printf("-1\n");
return 0;
}