[题目描述]
你知道黑暗城保有N个房间(1≤N≤1 000),M条可以制造的双向通道,以及每条通道的长度。
城堡是树形的并且满足下面的条件:如果所有的通道都被修建.设D[i]为第i号房间与第1号房间的最短路径长度;而S[i]为实际修建的树形城保中第i号房间与第1号房间的路径长度;要求对于所有整数i(1<=i<=N),有S[i]= D[i]成立。
你想知道有多少种不同的城保修建方案。当然,你只需要输出答案对2^31-1取模之后的结果就行了。
[样例输入]
4 6
1 2 1
1 3 2
1 4 3
2 3 1
2 4 2
3 4 1
[样例输出]
6
本题是最小生成树模板题,但需要需要用 Dijikstra 单源最短路。
具体代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int MAXN = 2147383647;
long long a[1011][1011];
bool b[1000011];
long long c[10011];
long long tot[10011];
int main()
{
memset (b, false, sizeof (b));
int n, m;
scanf ("%d%d", &n, &m);
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
{
if (i == j)
{
a[i][j] = 0;
}
else
{
a[i][j] = a[j][i] = MAXN;
}
}
}
int x, y, z;
for (int i = 1; i <= m; i++)
{
scanf ("%d%d%d", &x, &y, &z);
if (a[x][y] > z)
{
a[x][y] = a[y][x] = z;
}
}
for (int i = 1; i <= n; i++)
{
c[i] = a[1][i];
}
for (int i = 1; i <= n; i++)
{
long long minn = 2147483647;
int t;
for (int j = 1; j <= n; j++)
{
if (!b[j] && minn > c[j])
{
minn = c[j];
t = j;
}
}
b[t] = true;
for (int j = 1; j <= n; j++)
{
if (c[j] > a[t][j] + c[t])
{
c[j] = a[t][j] + c[t];
}
}
}
long long sum = 1;
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
{
if (i != j && c[j] == c[i] + a[i][j])
{
tot[j]++;
}
}
}
for (int i = 1; i <= n; i++)
{
if (tot[i] == 0)
{
continue;
}
sum *= tot[i];
sum %= 2147483647;
}
printf ("%d", sum);
return 0;
}