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