早上模拟赛考这题，结果竟然看错题目了orz
然后下午看完题解自己做的时候空间开小了白WA了好久orz
首先，如果以$S$为起点，一条边$(u,v)$在最短路上，则$dis[u]+edge[i]=dis[v]$
那么我们先以每个点为起点跑一遍最短路
每一次跑完最短路，对于一条边$i$，考虑它的经过次数
首先得满足上面那个条件，然后设$a[u]$表示从$S$走到$u$的最短路的方案，$v$表示经过$v$的最短路的方案
那么$ans[i]+=a[u]*b[v]$
$a$数组可以一遍拓扑排序顺便求出来，$b$数组可以用记忆化搜索搞出来
然后每一次跑完最短路都求出$a$和$b$，如果一条边出现在了这一次的最短路图中，就更新它的答案
1 //minamoto 2 #include<iostream> 3 #include<cstdio> 4 #include<cstring> 5 #include<queue> 6 #define ll long long 7 using namespace std; 8 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++) 9 char buf[1<<21],*p1=buf,*p2=buf; 10 template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;} 11 inline int read(){ 12 #define num ch-'0' 13 char ch;bool flag=0;int res; 14 while(!isdigit(ch=getc())) 15 (ch=='-')&&(flag=true); 16 for(res=num;isdigit(ch=getc());res=res*10+num); 17 (flag)&&(res=-res); 18 #undef num 19 return res; 20 } 21 const int N=1505,M=5005,mod=1e9+7; 22 struct node{ 23 int u,dis; 24 node(){} 25 node(int u,int dis):u(u),dis(dis){} 26 inline bool operator <(const node &b)const 27 {return dis>b.dis;} 28 }; 29 priority_queue<node> q; 30 int head[N],Next[M],ver[M],edge[M],from[M],tot,n; 31 ll ans[M],a[M],b[M];int dis[N],vis[M],d[N]; 32 int st[N],h,t,m; 33 inline void add(int u,int v,int e){ 34 ver[++tot]=v,Next[tot]=head[u],head[u]=tot,edge[tot]=e,from[tot]=u; 35 } 36 void dijkstra(int s){ 37 memset(dis,0x3f,sizeof(dis)); 38 memset(vis,0,sizeof(vis)); 39 dis[s]=0,q.push(node(s,0)); 40 while(!q.empty()){ 41 int u=q.top().u;q.pop(); 42 if(vis[u]) continue; 43 vis[u]=1; 44 for(int i=head[u];i;i=Next[i]){ 45 int v=ver[i]; 46 if(cmin(dis[v],dis[u]+edge[i])) 47 q.push(node(v,dis[v])); 48 } 49 } 50 } 51 inline void inita(int s){ 52 for(int u=1;u<=n;++u) 53 for(int i=head[u];i;i=Next[i]) 54 if(dis[ver[i]]==dis[u]+edge[i]) ++d[ver[i]]; 55 st[h=t=1]=s; 56 while(h<=t){ 57 int u=st[h++]; 58 for(int i=head[u];i;i=Next[i]){ 59 int v=ver[i]; 60 if(dis[v]==dis[u]+edge[i]){ 61 --d[v],(a[v]+=a[u])%=mod; 62 if(!d[v]) st[++t]=v; 63 } 64 } 65 } 66 } 67 void initb(int u){ 68 b[u]=1; 69 for(int i=head[u];i;i=Next[i]){ 70 int v=ver[i]; 71 if(dis[v]==dis[u]+edge[i]){ 72 vis[i]=1; 73 if(!b[v]) initb(v); 74 (b[u]+=b[v])%=mod; 75 } 76 } 77 } 78 int main(){ 79 // freopen("testdata.in","r",stdin); 80 n=read(),m=read(); 81 for(int i=1,u,v,e;i<=m;++i) 82 u=read(),v=read(),e=read(),add(u,v,e); 83 for(int i=1;i<=n;++i){ 84 dijkstra(i); 85 memset(a,0,sizeof(a)); 86 memset(b,0,sizeof(b)); 87 memset(vis,0,sizeof(vis)); 88 a[i]=1,inita(i),initb(i); 89 for(int j=1;j<=m;++j) 90 if(vis[j]) (ans[j]+=a[from[j]]*b[ver[j]])%=mod; 91 } 92 for(int i=1;i<=m;++i) printf("%lld ",ans[i]); 93 return 0; 94 }