题意:n个点之间有m条有向边,求从源点s到汇点t的最短路径有多少条,最短路径之间不能共用边。
利用前几天学的两个算法写了这题,感觉对两个算法还是不太熟练。
这题的关键还是在于建图。先用最短路径算法分别计算s和t点到各点的最短距离,若一条边的弧尾到s的距离+弧头到t的距离+边的权值=s到t的最短路径,那这条边必定在最短路径上。选择这些边建立一张新图,边的权值都为1,然后用最大流算法sap计算条数。
#include <iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<cmath>
#define INF 0x7fffffff
#define N 200100
using namespace std;
struct node
{
int to,next,w;
}e[N*8];
int n,cnt,head[N];
void init()
{
cnt=0;
memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{
e[cnt].to=v;
e[cnt].next=head[u];
e[cnt].w=w;
head[u]=cnt++;
e[cnt].to=u;
e[cnt].next=head[v];
e[cnt].w=0;
head[v]=cnt++;
}
int cur[N],gap[N],dis[N],pre[N];
int sap(int s,int t)
{
for(int i=1;i<=n;i++)
{
cur[i]=head[i];
gap[i]=dis[i]=0;
}
int ans=0,mmin=INF,u;
u=pre[s]=s;
gap[s]=n;
while(dis[s]<n)
{
int flag=0;
for(int &i=cur[u];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(dis[u]==dis[v]+1&&e[i].w)
{
flag=1;
mmin=min(mmin,e[i].w);
pre[v]=u;
u=v;
if(u==t)
{
ans+=mmin;
while(u!=s)
{
u=pre[u];
e[cur[u]].w-=mmin;
e[cur[u]^1].w+=mmin;
}
mmin=INF;
}
break;
}
}
if(flag) continue;
int md=n;
for(int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(e[i].w&&dis[v]<md)
{
md=dis[v];
cur[u]=i;
}
}
if(--gap[dis[u]]==0) return ans;
gap[dis[u]=md+1]++;
u=pre[u];
}
return ans;
}
void addedge(int u,int v,int w)
{
e[cnt].to=v;
e[cnt].next=head[u];
e[cnt].w=w;
head[u]=cnt++;
}
int d[N],vis[N],q[N];
void spfa(int u)
{
for(int i=1;i<=n;i++)
d[i]=INF,vis[i]=0;
d[u]=0;
vis[u]=1;
q[0]=u;
int k=0;
while(k>=0)
{
int t=q[k--];
vis[t]=0;
for(int i=head[t];i!=-1;i=e[i].next)
{
int c=e[i].to,w=e[i].w;
if(d[c]>d[t]+w)
{
d[c]=d[t]+w;
if(!vis[c]) vis[c]=1,q[++k]=c;
}
}
}
}
int d1[N],d2[N],u[N],v[N],w[N];
int main()
{
int T;
cin>>T;
while(T--)
{
int m;
cin>>n>>m;
init();
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u[i],&v[i],&w[i]);
addedge(u[i],v[i],w[i]);
}
int s,t;
cin>>s>>t;
spfa(s);
memcpy(d1,d,sizeof(d));
init();
for(int i=1;i<=m;i++)
addedge(v[i],u[i],w[i]);
spfa(t);
memcpy(d2,d,sizeof(d));
init();
for(int i=1;i<=m;i++)
if(u[i]!=v[i]&&d1[u[i]]+d2[v[i]]+w[i]==d1[t])
add(u[i],v[i],1);
cout<<sap(s,t)<<endl;
}
}