bzoj3661: Hungry Rabbit

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=3661

思路:

网络流,但这不是标算,bzoj为了大概是为了让网络流过才把时限开到100s

建两个源点,S向S‘连容量为K的边,这个是为了限制总流量的

每天的每只兔子拆成两个点,之间连1的边进行点容量限制,表示这只兔子今天只算一只(不限制后面会出问题)

每只兔子向下一天的自己连1的边,表示不用换人

每天设立两个中转点,之间连L的边,拆成两个也是为了限制流量

每只兔子向中转点连边,中转点再向下一天的每只兔子连边,表示可以换L只兔子

点数1280000+,边数不超过5000000

然后就没有然后了

正解与污误解的分界线


#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
const int N=820,maxn=1500010,maxm=5000010,inf=1061109567;
using namespace std;
int n,m,K,L,map[N][N],rab[N][N][2],cnt,lim[N][2],Edge[N][N];char ch[N];

struct Flow{
	int pre[maxm],now[maxn],son[maxm],val[maxm],tot,dis[maxn],S,SS,T,q[maxm+10],head,tail;
	void add(int a,int b,int c){pre[++tot]=now[a],now[a]=tot,son[tot]=b,val[tot]=c;}
	void ins(int a,int b,int c){add(a,b,c),add(b,a,0);}
	void init(){memset(now,0,sizeof(now)),tot=1,S=maxn-3,SS=maxn-2,T=maxn-1;}
	bool bfs(){
		memset(dis,-1,sizeof(dis));
		q[tail=1]=S,dis[S]=head=0;
		while (head!=tail){
			if (++head>maxn) head=1;
			int x=q[head];
			for (int y=now[x];y;y=pre[y])
				if (dis[son[y]]==-1&&val[y]){
					if (++tail>maxn) tail=1;
					dis[son[y]]=dis[x]+1,q[tail]=son[y];
				}
		}
		return dis[T]>0;
	}
	int find(int x,int low){
		if (x==T) return low;
		int y,res=0;
		for (y=now[x];y;y=pre[y]){
			if (dis[son[y]]!=dis[x]+1||!val[y]) continue;
			int tmp=find(son[y],min(low,val[y]));
			res+=tmp,low-=tmp,val[y]-=tmp,val[y^1]+=tmp;
			if (!low) break;
		}
		if (!y) dis[x]=-1;
		return res;
	}
	void build(){
		ins(S,SS,K);	
		for (int i=1;i<=m;i++)
			for (int j=1;j<=n;j++){
				ins(rab[i][j][0],rab[i][j][1],1);//点容量限制
				Edge[i][j]=tot-1;
			}
		for (int i=1;i<=n;i++) if (map[1][i]) ins(SS,rab[1][i][0],1);
		for (int i=2;i<=m;i++)
			for (int j=1;j<=n;j++)
				if (map[i][j]) ins(rab[i-1][j][1],rab[i][j][0],1);//向下一天兔子连边
		for (int i=1;i<m;i++)
			for (int j=1;j<=n;j++)
				if (map[i][j]) ins(rab[i][j][1],lim[i][0],1);
		for (int i=1;i<m;i++) ins(lim[i][0],lim[i][1],L);
		for (int i=1;i<m;i++)
			for (int j=1;j<=n;j++)
				if (map[i+1][j]) ins(lim[i][1],rab[i+1][j][0],1);
		for (int i=1;i<=n;i++) ins(rab[m][i][1],T,1);
	}
	void work(){
		int res=0;
		while (bfs()) res+=find(S,inf);
		if (res!=K){puts("-1");return;}

		for (int i=1;i<=m;i++,puts(""))
			for (int j=1;j<=n;j++)
				if (!val[Edge[i][j]]) printf("%d ",j);
	}
}F;

int main(){
	scanf("%d%d%d%d",&n,&m,&K,&L);F.init();
	for (int i=1;i<=n;i++){
		scanf("%s",ch+1);
		for (int j=1;j<=m;j++) map[j][i]=ch[j]-'0',rab[j][i][0]=++cnt,rab[j][i][1]=++cnt;
	}
	for (int i=1;i<=m;i++) lim[i][0]=++cnt,lim[i][1]=++cnt;
	F.build(),F.work();
	return 0;
}



你可能感兴趣的:(bzoj3661: Hungry Rabbit)