Description
C国有n座城市,城市之间通过m条单向道路连接。一条路径被称为最短路,当且仅当不存在从它的起点到终点的另外一条路径总长度比它小。两条最短路不同,当且仅当它们包含的道路序列不同。我们需要对每条道路的重要性进行评估,评估方式为计算有多少条不同的最短路经过该道路。现在,这个任务交给了你。
Input
第一行包含两个正整数n、m
接下来m行每行包含三个正整数u、v、w,表示有一条从u到v长度为w的道路
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
HINT
数据规模
30%的数据满足:n≤15、m≤30
60%的数据满足:n≤300、m≤1000
100%的数据满足:n≤1500、m≤5000、w≤10000
一道最短路的变形题
如果一条边i在最短路上,那么i的贡献=当前起点到这条边的起点的最短路数*当前起点到所有点的最短路中经过当前变的终点的数目。
那么设当前起点到这条边的起点的最短路数为st[ ],当前起点到所有点的最短路中经过当前变的终点的数目为ed[ ],然后每个点在最短路上的前驱为pre[ ],对于每个点跑一边spfa,然后每次求出每个点的st,ed,pre,ans+=st[u]*ed[v]。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define N 1510
#define M 5010
#define mod 1000000007
using namespace std;
int n,m,head[N],dis[N],pre[N],st[M],ed[M],ans[M];
bool used[N];
struct Node{
int to,nxt,w,frm;
}edge[M];
void add(int j,int x,int y,int z)
{
edge[j].nxt=head[x];
edge[j].to=y;
edge[j].frm=x;
edge[j].w=z;
head[x]=j;
}
void SPFA(int s)
{
memset(dis,0x3f,sizeof dis); memset(used,0,sizeof used);
queue<int> q;
dis[s]=0;
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
used[u]=0;
for(int i=head[u];i;i=edge[i].nxt)
{
int v=edge[i].to;
if(dis[v]>dis[u]+edge[i].w)
{
dis[v]=dis[u]+edge[i].w;
if(!used[v])
{
used[v]=1;
q.push(v);
}
}
}
}
}
void get_pre(int x)
{
used[x]=1;
for(int i=head[x];i;i=edge[i].nxt)
{
int v=edge[i].to;
if(dis[v]==dis[x]+edge[i].w)
{
pre[v]++;
if(!used[v]) get_pre(v);
}
}
}
void get_st(int x)
{
for(int i=head[x];i;i=edge[i].nxt)
{
int v=edge[i].to;
if(dis[v]==dis[x]+edge[i].w)
{
st[v]+=st[x];
st[v]%=mod;
pre[v]--;
if(!pre[v]) get_st(v);
}
}
}
void get_ed(int x)
{
ed[x]=1;
for(int i=head[x];i;i=edge[i].nxt)
{
int v=edge[i].to;
if(dis[v]==dis[x]+edge[i].w)
{
if(!ed[v]) get_ed(v);
ed[x]+=ed[v];
ed[x]%=mod;
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(i,x,y,z);
}
for(int i=1;i<=n;i++)
{
SPFA(i);
memset(used,0,sizeof used);
memset(st,0,sizeof st);
memset(ed,0,sizeof ed);
get_pre(i);
st[i]=1; get_st(i);
get_ed(i);
for(int j=1;j<=m;j++)
{
int u=edge[j].frm,v=edge[j].to;
if(dis[v]==dis[u]+edge[j].w)
{
ans[j]+=st[u]*ed[v];
ans[j]%=mod;
}
}
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
return 0;
}