通过学习我们可以发现原题目要求的树结构就是“最短路径生成树”,什么是最短路径生成树呢?就是一张图的源点到图中的所有结点的最短路径构成的树,为什么他满足树结构呢?因为假设有n个点,因为是最短路径一定只有n-1条边把他连起来。
那对于本题,每个点的 D [ i ] D[i] D[i]可以先跑一遍 D i j k s t r a Dijkstra Dijkstra得出来,接下来思考如何计数。
计数方法明显是要使用乘法原理计数,也就是说我们可以得出每一步的方案数再乘进答案中。
接下来考虑如何的出每一步的方案数,所谓方案数也就是对于每一个 D [ i ] D[i] D[i]可以从多少个 D [ j ] D[j] D[j]转移过来,那么显然,比较大的距离只能从比较小的距离转移过来。
所以我们可以先给 D D D数组递增排序。 这样比 i i i小的都是有可能转移到 i i i的点,那么我们只需要判断一下他们之间有没有边可以转移到即可。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#define mkp(a,b) make_pair(a,b)
using namespace std;
typedef long long ll;
typedef pair<int,int>pii;
const int N=1e3+10;
const int M=5e5+10;
const ll mod=(1ll<<31)-1;
ll mp[N][N];
int n,m;
struct edge{int x,y,c,next;}a[M*2]; int len,last[N];
void ins(int x,int y,int c)
{
a[++len].x=x;a[len].y=y;a[len].c=c;
a[len].next=last[x];last[x]=len;
}
priority_queue<pii,vector<pii>,greater<pii> > q;
int d[N]; bool v[N];
void dijkstra()
{
memset(d,63,sizeof(d)); d[1]=0;
memset(v,0,sizeof(v));
q.push(mkp(0,1));
while(!q.empty())
{
int x=q.top().second; q.pop();
if(v[x]) continue;
v[x]=true;
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if(d[y]>d[x]+a[k].c)
{
d[y]=d[x]+a[k].c;
q.push(mkp(d[y],y));
}
}
}
}
struct node
{
int id,d;
}b[N];
bool cmp(node a,node b){return a.d<b.d;}
int main()
{
// freopen("a.in","r",stdin);
// freopen("a.out","w",stdout);
scanf("%d%d",&n,&m);
memset(mp,63,sizeof(mp));
len=0; memset(last,0,sizeof(last));
for(int i=1;i<=m;i++)
{
int x,y,c;scanf("%d%d%d",&x,&y,&c);
ins(x,y,c); ins(y,x,c);
mp[x][y]=mp[y][x]=c;
}
dijkstra();
for(int i=1;i<=n;i++) b[i].id=i,b[i].d=d[i];
sort(b+1,b+n+1,cmp);
ll ans=1;
for(int i=2;i<=n;i++)
{
ll cnt=0;
for(int j=i-1;j>=1;j--)
{
int y=b[i].id,x=b[j].id;
if(d[y]==d[x]+mp[x][y]) cnt++;
}
ans=ans*cnt%mod;
}
printf("%lld\n",ans);
return 0;
}