#10064. 「一本通 3.1 例 1」黑暗城堡

题目描述

你知道黑暗城堡有 N个房间，M条可以制造的双向通道，以及每条通道的长度。

城堡是树形的并且满足下面的条件：

设 Di 为如果所有的通道都被修建，第 i 号房间与第 1 号房间的最短路径长度；

而 Si 为实际修建的树形城堡中第 i 号房间与第 1 号房间的路径长度；

要求对于所有整数 i (1<=i<=N)，有 Si = Di 成立。

你想知道有多少种不同的城堡修建方案。当然，你只需要输出答案对 2^31-1 取模之后的结果就行了。

输入格式

第一行为两个由空格隔开的整数 N , M ;

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

 输出格式

一个整数：不同的城堡修建方案数对 2^31-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^31-1 取模之后的结果为 6。

数据范围与提示

对于全部数据，1<=N<=1000，1<=M<=N*(N-1)/2，1<=l<=200。

意思就是按最小的成本来修，有多少种方法；也就是最小的路径可以有多少种走法：

做法就是：先用 dinkstra 求一个单源最短路，然后再遍历一遍，每个点加入树中和最短路的长度相等的数目，相乘；

得到的再取余就是结果。

代码如下：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
using namespace std;

typedef long long ll;
const int maxn=1e3+5;
const int inf=0x3f;
int u,n,m,minn;
const ll mod=(1<<31)-1;

int e[maxn][maxn];
int dis[maxn];
int vis[maxn];



void dijkstra()
{
   memset(dis,inf,sizeof(dis));
   memset(vis,0,sizeof(vis));

    for(int i=1;i<=n;i++)
    {
        dis[i]=e[1][i];
    }
    dis[1]=0;
        for(int i=1;i<=n-1;i++)
        {
            minn=inf;
            for(int j=1;j<=n;j++)
            {

                if(vis[j]==0&&dis[j]<minn)
                {

                    minn=dis[j];
                    u=j;
                }
            }
            vis[u]=1;
            for(int k=1;k<=n;k++)
            {

                if(e[u][k]<inf)
                {

                    dis[k]=min(dis[k],dis[u]+e[u][k]);
                }
            }
        }
}




int main()
{

    int x,y,z;
    scanf("%d %d",&n,&m);
    memset(e,inf,sizeof(e));
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&x,&y,&z);
        e[x][y]=e[y][x]=min(e[x][y],z);
    }
    dijkstra();
    ll ans=1;
    for(int i=2;i<=n;i++)
    {
        int constt=0;
        for(int j=1;j<=n;j++)
            if(e[j][i]!=inf)
            if(dis[i]==dis[j]+e[j][i])
            constt++;
           // printf("\t\t%d\n",constt);
        ans=(ans*constt)%mod;
        //printf("%lld\n",ans);
    }
    printf("%lld\n",ans);

}