黑暗城堡

[题目描述]
你知道黑暗城保有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;
}