poj 1815 Friendship 最小割输出最小方案

这题卡了好久啊,最小割模型很容易想,拆点就行。就像poj的Thieves一样

每个点 a拆成 a->a',容量为1。

其他相连的点 a'->b ,容量为INF

源连接s',t连接汇


问题在于输出最小的割集

更好的方法我还不会,只能枚举。

这里基于贪心的思想,从小到大删边,

若删除i->i',会使得最小割变小,则输出i,并且i->i'这条边不要恢复

若最小割不变,则恢复这条边,继续枚举。

一开始就是因为恢复了要割去的边,无限WA。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<string>
#define eps  1e-12
#define INF   0x7fffffff
#define maxn 22222
using namespace std;
int n,m;
int en;
int st,ed,s,t;	//源点和汇点
int dis[maxn] ;//dis[i],表示  到 原点  s 的 层数
int que[9999999];
bool delet[maxn];
struct edge
{
	int u,to,c,next;
};
edge e[999999];
edge save[999999];
int head[maxn];
void add(int a,int b,int c)
{
	e[en].to=b;
	e[en].c=c;
	e[en].next=head[a];
	head[a]=en++;

	e[en].to=a;
	e[en].c=0;
	e[en].next=head[b];
	head[b]=en++;
}
int bfs()
{
    memset(dis,-1,sizeof(dis));
    dis[st]=0;
    int front=0,rear=0;
    que[rear++]=st;
    while(front<rear)
    {
        int j=que[front++];
        for(int k=head[j];k!=-1;k=e[k].next)
        {
            int i=e[k].to;
			if(dis[i]==-1&&e[k].c)
            {
                dis[i] = dis[j]+ 1 ;
                que[rear++]=i;
                if(i==ed) return true;
            }
        }
    }
    return false;
}
int dfs(int x,int mx)
{
    int i,a;
    if(x==ed) return mx ;
    int ret=0;
    for(int k=head[x];k!=-1&&ret<mx;k=e[k].next)
    {
        if(e[k].c&&dis[e[k].to]==dis[x]+1)
        {
            int dd=dfs(e[k].to,min(e[k].c,mx-ret));
            e[k].c-=dd;
            e[k^1].c+=dd;
            ret+=dd;
        }
    }
    if(!ret) dis[x]=-1;
    return ret;
}
bool del[maxn];
int a[205][205];
void init()
{
	st=0;     //源
    ed=n+n+1;     //汇
	for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    memset(del,0,sizeof(del));
}
void pr(int a,int b,int c)
{
    printf("%d --> %d =%d \n",a,b,c);
}
void build()
{
    en=0;
    memset(head,-1,sizeof(head));
    add(st,s+n,INF);
    for(int i=1;i<=n;i++)
    {
        if(!del[i])
        {
            add(i,i+n,1);
        }
        for(int j=1;j<=n;j++)
        {
            if(a[i][j])
            {
                add(i+n,j,INF);
            }
        }
    }
    add(t,ed,INF);
}
int dinic()
{
    int tmp=0;
    int maxflow=0;
    while(bfs())
    {
        while(tmp=dfs(st,INF)) maxflow+=tmp;
    }
    return maxflow;
}
int main()
{
    while(scanf("%d%d%d",&n,&s,&t)!=EOF)
    {
        init();
        build();
        if(a[s][t]) {puts("NO ANSWER!");continue;}
        int ans=dinic();
        printf("%d\n",ans);
        for(int i=1;i<=n;i++)
        {
            if(i==s||i==t) continue;
            del[i]=1;
            build();
            int tmp=dinic();
            if(tmp<ans)
            {
                if(ans-1) printf("%d ",i);
                else printf("%d",i);
                ans--;
            }
            else del[i]=0;
            if(ans==0) break;
        }
        puts("");
    }
}
/*
3 1 3
1 1 0
1 1 1
0 1 1

10 2 1
0 0 1 1 1 0 1 1 1 0
0 0 1 1 0 0 0 1 0 1
1 1 0 1 0 1 0 1 0 1
1 1 1 0 0 0 1 0 0 1
1 0 0 0 0 0 1 0 1 0
0 0 1 0 0 0 0 0 0 1
1 0 0 1 1 0 0 1 0 1
1 1 1 0 0 0 1 0 0 1
1 0 0 0 1 0 0 0 0 1
0 1 1 1 0 1 1 1 1 0

10 10 1
0 1 1 1 1 0 1 1 0 0
1 0 0 0 0 1 1 1 1 1
1 0 0 1 1 1 1 1 1 1
1 0 1 0 1 1 0 1 1 0
1 0 1 1 0 0 1 1 1 1
0 1 1 1 0 0 1 0 0 1
1 1 1 0 1 1 0 1 1 0
1 1 1 1 1 0 1 0 0 0
0 1 1 1 1 0 1 0 0 1
0 1 1 0 1 1 0 0 1 0
*/


你可能感兴趣的:(poj 1815 Friendship 最小割输出最小方案)