【最短路生成树】【例题】【ybt3.1.1】黑暗城堡

LOJ10064

题目：

求有多少种方案使$1->i$路径长为原始最短路$di{s}_i$

题解：

拿到题，要用$di{s}_i$，那就先跑一遍dijkstra
然后就蒙圈了，怎么求方案数？
我们想到了换路，将1->j的路径换成多个1->k,k->j的路径
程序中就是mp[x][y]（直接路径）+dis[y]==dis[x]
额，似乎乘法原理，累计换$j$的方案后累乘，没了？！
$Tips:$由于跑完最短路后$di{s}_k>di{s}_j$，那么此$k$绝不会会用来为j换路径（做贡献），于是按$dis$排一下序，每次看$j$之前的$k$进行更行

#include<bits/stdc++.h>
#define ll long long
#define dis(x) D[x].dis
using namespace std;
const int N=1e3+10,M=2e6+10,mod=2147483647;
int n,m;
int head[N],nex[M],to[M],val[M],tot;
void build(int u,int v,int w)
{
	tot++;nex[tot]=head[u];to[tot]=v;val[tot]=w;head[u]=tot;
}
struct qu{
	int dis,id;
	bool operator < (const qu x)const{return dis<x.dis;}
}D[N];
bool vis[N];
priority_queue<pair<int,int> >q;
void dijkstra()
{
	for(int i=1;i<=n;i++)D[i].id=i,dis(i)=1e9;
	q.push(make_pair(0,1));dis(1)=0;
	while(!q.empty())
	{
		int u=q.top().second;q.pop();
		if(vis[u])continue;vis[u]=1;
		for(int i=head[u];i;i=nex[i])
		{
			int v=to[i];
			if(dis(v)>dis(u)+val[i])
			{
				dis(v)=dis(u)+val[i];
				q.push(make_pair(-dis(v),v));
			}
		}
	}
}
ll ans=1;
int mp[N][N];
void d_p()
{
	sort(D+1,D+n+1);
	for(int i=2;i<=n;i++)
	{
		ll tmp=0;
		int x=D[i].id;
		for(int j=1;j<i;j++)
		{
			int y=D[j].id;
			if(mp[x][y]+dis(j)==dis(i))tmp++;
		}
		ans=ans*tmp%mod;
	}
}
int main()
{
	//freopen("1.in","r",stdin);
	//freopen("1.ans","w",stdout);
	memset(mp,0x7f,sizeof(mp));
	scanf("%d%d",&n,&m);
	for(int u,v,w,i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		build(u,v,w);build(v,u,w);
		mp[u][v]=mp[v][u]=w;
	}
	dijkstra();
	d_p();
	printf("%lld",ans);
}