[HAOI2012]Road

Description
C国有n座城市,城市之间通过m条单向道路连接。一条路径被称为最短路,当且仅当不存在从它的起点到终点的另外一条路径总长度比它小。两条最短路不同,当且仅当它们包含的道路序列不同。我们需要对每条道路的重要性进行评估,评估方式为计算有多少条不同的最短路经过该道路。现在,这个任务交给了你。

Input
第一行包含两个正整数n、m
接下来m行每行包含三个正整数u、v、w,表示有一条从u到v长度为w的道路
n≤1500、m≤5000、w≤10000

Output
输出应有m行,第i行包含一个数,代表经过第i条道路的最短路的数目对1000000007取模后的结果

Sample Input
4 4
1 2 5
2 3 5
3 4 5
1 4 8

Sample Output
2
3
2
1


首先对于每个点求一次SPFA或dijkstra,然后标记出最短路上的边,对其进行拓扑排序,求出一条边的出点能到达的点能经过的最短路的个数,能到达这条边的入点的最短路的个数,然后使用乘法原理即可

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline int read(){
	int x=0,f=1;char ch=getchar();
	for (;ch<'0'||ch>'9';ch=getchar())	if (ch=='-')    f=-1;
	for (;ch>='0'&&ch<='9';ch=getchar())	x=(x<<1)+(x<<3)+ch-'0';
	return x*f;
}
inline void print(int x){
	if (x>=10)     print(x/10);
	putchar(x%10+'0');
}
const int N=1.5e3,M=5e3,mod=1e9+7;
struct S1{
	#define ls (p<<1)
	#define rs (p<<1|1)
	#define fa (p>>1)
	struct node{
		int x,v;
		bool operator <(const node &a)const{return v<a.v;}
	}Q[N+10];
	int tot;
	void clear(){tot=0;}
	void insert(int x,int v){
		Q[++tot]=(node){x,v};
		int p=tot;
		while (p!=1&&Q[p]<Q[fa])	swap(Q[p],Q[fa]),p=fa;
	}
	void Delete(){
		Q[1]=Q[tot--];
		int p=1,son;
		while (ls<=tot){
			if (rs>tot||Q[ls]<Q[rs])	son=ls;
			else	son=rs;
			if (Q[son]<Q[p])	swap(Q[p],Q[son]),p=son;
			else	break;
		}
	}
}Heap;
int pre[M+10],now[N+10],child[M+10],val[M+10],from[M+10];
int deep[N+10],cnt[N+10],h[N+10];
int Ans[M+10],cnt1[N+10],cnt2[N+10];
bool vis[N+10],can[M+10];
int n,m,tot;
void join(int x,int y,int z){pre[++tot]=now[x],now[x]=tot,from[tot]=x,child[tot]=y,val[tot]=z;}
void dijkstra(int x){
	memset(vis,0,sizeof(vis));
	memset(deep,63,sizeof(deep));
	Heap.clear(),Heap.insert(x,deep[x]=0);
	while (Heap.tot){
		int Now=Heap.Q[1].x;
		Heap.Delete();
		if (vis[Now])	continue;
		vis[Now]=1;
		for (int p=now[Now],son=child[p];p;p=pre[p],son=child[p]){
			if (deep[son]>deep[Now]+val[p]){
				deep[son]=deep[Now]+val[p];
				Heap.insert(son,deep[son]);
			}
		}
	}
	memset(can,0,sizeof(can));
	//记录最短路上的边
	for (int i=1;i<=m;i++)	if (deep[from[i]]+val[i]==deep[child[i]])	can[i]=1;
}
void topo(int x){//拓扑排序
	memset(cnt,0,sizeof(cnt));
	memset(cnt1,0,sizeof(cnt1));
	memset(cnt2,0,sizeof(cnt2));
	for (int i=1;i<=m;i++)	if (can[i])	cnt[child[i]]++;
	int head=1,tail=1;
	cnt1[h[1]=x]=1;
	for (;head<=tail;head++){
		int Now=h[head];
		for (int p=now[Now],son=child[p];p;p=pre[p],son=child[p]){
			if (!can[p])	continue;
			cnt1[son]=(cnt1[son]+cnt1[Now])%mod;
			if (!(--cnt[son]))	h[++tail]=son;
		}
	}
	for (int i=tail,Now;i;i--){
		cnt2[Now=h[i]]++;
		for (int p=now[Now],son=child[p];p;p=pre[p],son=child[p]){
			if (!can[p])	continue;
			cnt2[Now]=(cnt2[Now]+cnt2[son])%mod;
		}
	}
}
void work(int x){
	dijkstra(x),topo(x);
	for (int i=1;i<=m;i++)	if (can[i])	Ans[i]=(Ans[i]+1ll*cnt1[from[i]]*cnt2[child[i]]%mod)%mod;
}
int main(){
	n=read(),m=read();
	for (int i=1;i<=m;i++){
		int x=read(),y=read(),z=read();
		join(x,y,z);
	}
	for (int i=1;i<=n;i++)	work(i);
	for (int i=1;i<=m;i++)	printf("%d
",Ans[i]);
	return 0;
}
原文地址:https://www.cnblogs.com/Wolfycz/p/9744333.html