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

你可能感兴趣的:(最短路,OJ题解)