黑暗城堡(dijkstra算法)

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld
题目描述
你知道黑暗城堡有N个房间，M条可以制造的双向通道，以及每条通道的长度。
城堡是树形的并且满足下面的条件：
设D _i为如果所有的通道都被修建，第i号房间与第1号房间的最短路径长度；
而S_i为实际修建的树形城堡中第i号房间与第1号房间的路径长度；
要求对于所有整数i(1≤i≤N)，有S_i=D_i成立。
你想知道有多少种不同的城堡修建方案。当然，你只需要输出答案对2³¹−1取模之后的结果就行了。
输入描述:
输入格式
第一行为两个由空格隔开的整数N,M;
第二行到第M+1行为3个由空格隔开的整数x,y,l：表示x号房间与y号房间之间的通道长度为l。
输出描述:
一个整数：不同的城堡修建方案数对2³¹−1取模之后的结果。
示例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。
而不同的城堡修建方案数对2³¹−1取模之后的结果为6。

解题思路：
解题步骤如下，看不懂欢迎留言哦
1、dijkstra算法，算出各点最短路径
2、用d[j]==d[i]+c[i][j]?判断是否为子结点，即i点的入度
3、逆向思考，因为重建一个结点只需要一条路径，如果重建一个结点的最短路径有多条，你可以选择任意一条来修建，所以总数就是各点入度的乘积。

代码：

#include<iostream>
#define INT 0x3f3f3f3f
using namespace std;
int main()
{
    int a,b,c[1010][1010],d[1010],vis[1010]={0},e,f,g,i,j,ans=0;
    //c[][]记录点到点的长度，d[]纪录各点到起点1的长度,vis[]标记
    long long sum=1;
    for(i=0;i<1010;i++)//初始化
        for(j=0;j<1010;j++)
        {
            c[i][j]=INT;
            d[j]=INT;
        }
    cin>>a>>b;
    while(b--)//dijkdtra算法
    {
        cin>>e>>f>>g;
        c[e][f]=c[f][e]=g;
    }
    d[1]=0;
    for(i=1;i<=a;i++)
    {
        e=0;
        for(j=1;j<=a;j++)
            if(!vis[j]&&(e==0||d[j]<=d[e]))
                e=j;
        vis[e]=1;
        for(j=1;j<=a;j++)
            if(!vis[j])
                d[j]=min(d[j],d[e]+c[e][j]);
    }
    for(i=2;i<=a;i++)//算总数
    {
        ans=0;
        for(j=1;j<=a;j++)
            if(d[i]==d[j]+c[j][i]&&d[i]!=INT)//算出入度个数
                ans++;
        sum=sum*ans%((1<<31)-1);//乘法
    }
    cout<<sum<<endl;
}