计蒜之道 2016 复赛 F.菜鸟物流的运输网络

考试时做的时间最长的一道题...

各种细节判定

换用网络流 T了一些..没跳出来 但大致思路都是对的!!

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int MAXN=100+10;
const int MAXM=800*100+10;

int tot=1,g[MAXN],nnext[MAXM],num[MAXM],ix[MAXM],iy[MAXM],flow[MAXM];
void add(int x,int y,int z)
{
// cout<<x<<' '<<y<<' '<<z<<endl;
tot++;
nnext[tot]=g[x];
g[x]=tot;
num[tot]=y;
flow[tot]=z;
ix[tot]=x;
iy[tot]=y;
}
int team[MAXN],head,tail;
bool used[MAXM];
bool can[MAXN];
int d[MAXN*2];
int S,T,mid,s,t;
int n,m;
bool bfs()
{
memset(d,0,sizeof(d));
head=tail=0;
team[++tail]=S;
d[S]=1;
while(head<tail)
{
int x=team[++head];
// cout<<x<<"*"<<d[x]<<endl;
for(int i=g[x];i;i=nnext[i])
{
int tmp=num[i];
if(flow[i]>=1&&d[tmp]==0)
{
d[tmp]=d[x]+1;
team[++tail]=tmp;
}
}
}
// cout<<T<<endl;
if(d[T]==0) return false;
return true;
}
int dfs(int x,int mmin)
{
if(x==T) return mmin;
int tmp,f=0;
for(int i=g[x];i;i=nnext[i])
if(d[num[i]]==d[x]+1&&flow[i]>=1&&(tmp=dfs(num[i],min(mmin,flow[i]))))
{
flow[i]-=tmp;
flow[i^1]+=tmp;
used[i]^=1;
used[i^1]^=1;
f+=tmp;
mmin-=tmp;
if(mmin==0) return f;
}
d[x]=0;
return f;
}

void dfs_w(int x,int fa)
{
if(x==mid+n) printf("%d ",mid) ;
// cout<<'*'<<x<<endl;
if(x!=t)
{if(x<=n) printf("%d ",x);}
else
{
printf("%d\n",x);
return ;
}
for(int i=g[x];i;i=nnext[i])
if(used[i]&&num[i]!=T&&fa!=num[i]/*&can[num[i]]&&(flow[i]==c)*/)
{
//cout<<"*"<<flow[i]<<' '<<flow[i^1]<<endl;
dfs_w(num[i],x);return ;
}
}
int main()
{
int TT;cin>>TT;
while(TT--)
{
tot=1;
memset(g,0,sizeof(g));
memset(used,false,sizeof(used));
memset(can,false,sizeof(can));
scanf("%d %d %d %d %d",&n,&m,&s,&t,&mid);
for(int i=1;i<=n;i++) if(i!=mid) add(i,n+i,1),add(n+i,i,0);
add(mid,mid+n,0);add(mid+n,mid,1);
for(int x,y,i=1;i<=m;i++)
{
scanf("%d %d",&x,&y);
add(x+n,y,1);
add(y,x+n,0);
add(y+n,x,1);
add(x,y+n,0);
}
T=2*n+1;
S=mid+n;
add(s+n,T,1);add(T,s+n,0);
add(t+n,T,1);add(T,t+n,0);

int sum=0;
while(bfs()&&sum!=2) sum+=dfs(S,1);
// cout<<sum<<endl;
/* for(int i=1;i<=tot;i++)
if(used[i]==true)
/*cout<<ix[i]<<" "<<iy[i]<<endl,*//*can[ix[i]]=true,can[iy[i]]=true;*/

dfs_w(s,0);
}
return 0;
}