最短路径生成树的数量-黑暗城堡

信息学奥赛一本通T1486-黑暗城堡

时间限制: 2s 内存限制: 192MB 提交: 18 解决: 9

题目描述

知道黑暗城堡有 N 个房间，M 条可以制造的双向通道，以及每条通道的长度。
城堡是树形的并且满足下面的条件：
设 Di为如果所有的通道都被修建，第 i 号房间与第 1 号房间的最短路径长度；
而 Si 为实际修建的树形城堡中第 i 号房间与第 1号房间的路径长度；
要求对于所有整数 i(1≤i≤N)，有 Si=Di成立。
你想知道有多少种不同的城堡修建方案。当然，你只需要输出答案对 231−1 取模之后的结果就行了。

输入格式

第一行为两个由空格隔开的整数 N,M;
第二行到第 M+1 行为3 个由空格隔开的整数 x,y,l：表示 x 号房间与 y 号房间之间的通道长度为 l。

输出格式

一个整数：不同的城堡修建方案数对 231−1 取模之后的结果。

样例输入

复制

4 6
1 2 1
1 3 2
1 4 3
2 3 1
2 4 2
3 4 1

样例输出

复制

6

提示

样例说明
一共有 4 个房间，6 条道路，其中 1 号和 2 号，1 号和 3 号，1 号和 4 号，2 号和 3 号，2 号和 4 号，3 号和 4 号房间之间的通道长度分别为 1，2，3，1，2，1。
而不同的城堡修建方案数对 231−1 取模之后的结果为 6。
数据范围:
对于全部数据，1≤N≤1000，1≤M≤N(N−1)/2,1≤l≤200。

 题解：

思路为先用dijkstra求得1到各点最短路径，再依次通过dis[j]=dis[u]+w[i],即若该点的dis值加上到下一个点的权值等于下一个点的dis值，那么到下一个点的路径数加一。最后将所以的点的路径数相乘取mod即答案，注意取模时千万别用2e31-1，这个会有问题，直接用整型，别用浮点型，至于为什么希望有大佬帮忙回复一下，感谢。

#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int>PII;
const int N = 1010, M = N * N;
const long long mod = (1<<31)-1;
int h[N], e[M], ne[M], w[M], idx;
int dis[N];
int cnt[N];
bool st[N];
int n, m;

void add(int a, int b, int c) {
	e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}

void dij()
{
	memset(dis, 0x3f, sizeof(dis));
	priority_queue<PII, vector<PII>, greater<PII>>q;
	q.push({ 0,1 });
	dis[1] = 0;
	while (q.size()) {
		auto t = q.top();
		q.pop();
		int u = t.second;
		if (st[u]) {
			continue;
		}
		st[u] = true;
		for (int i = h[u];~i;i = ne[i]) {
			int j = e[i];
			if (dis[j] > dis[u] + w[i]) {
				dis[j] = dis[u] + w[i];
				q.push({ dis[j],j });
			}
		}
	}

}
void get_cnt()
{
	for (int u = 1;u <= n;u++) {
		for (int i = h[u];~i;i = ne[i]) {
			int j = e[i];
			if (dis[j] == dis[u] + w[i]) {
				cnt[j]++;
			}
		}
	}
}
long long get_ans()
{
	long long ans = 1;
	cnt[1] = 1;
	for (int i = 1;i <= n;i++) {
		ans *= cnt[i];
		ans %= mod;
	}
	return ans;
}
int main()
{
	cin >> n >> m;
	memset(h, -1, sizeof h);
	for (int i = 1;i <= m;i++) {
		int a, b, c;
		cin >> a >> b >> c;
		add(a, b, c);
		add(b, a, c);
	}
	dij();
	get_cnt();
	long long res = get_ans();
	cout << res;
}