POJ 1847 Tram 【最短路spfa 模板题】

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