POJ3189 Steady Cow Assignment 二分枚举+dinic最大流

但是不幸的是,一直超时。

查不出来是不是我哪写错了。

还是难道非得sap?

#include<iostream>
using namespace std;
#define MIN(a,b) (a<b?a:b)
#define MAX(a,b) (a>b?a:b)
const int N=1005;
const int B=25;
int n,b;
const int inf=99999999;
struct Edge
{
	int v,next,w,re;
}edge[N*B*2+N*2+4];
int edgehead[N+B];
int k=1;
int mat[N][B];
int barn[B];
int level[N+B];
bool visit[N+B];
int que[N+B];
int final;
void addedge(int u,int v,int w)
{
	edge[k].v=v;
	edge[k].w=w;
	edge[k].re=k+1;
	edge[k].next=edgehead[u];
	edgehead[u]=k++;
	edge[k].v=u;
	edge[k].w=0;
	edge[k].re=k-1;
	edge[k].next=edgehead[v];
	edgehead[v]=k++;
}
bool bfs()
{
	memset(visit,0,sizeof(visit));
	memset(level,0,sizeof(level));
	int head=1,tail=1;
	visit[0]=true;
	level[0]=0;
	que[tail++]=0;
	while(head<tail)
	{
		int now=que[head++];
		if(now==n+b+1)
		{
			return true;
		}
		for(int i=edgehead[now];i!=0;i=edge[i].next)
		{
			int to=edge[i].v;
			if(!visit[to]&&edge[i].w>0)
			{
				que[tail++]=to;
				visit[to]=true;
				level[to]=level[now]+1;
			}
		}
	}
	return false;
}

int  dinic(int now,int sum)
{
	if(now==final)
		return sum;//因为是汇点了,return sum则说明流全部消耗。
	int os=sum;
	for(int i=edgehead[now];sum>0&&i!=0;i=edge[i].next)//notice sum>0
	{
		int to=edge[i].v;
		if(level[to]==level[now]+1&&edge[i].w)
		{
			int ret=dinic(to,MIN(sum,edge[i].w));//notice MIN
			edge[i].w-=ret;
			edge[edge[i].re].w+=ret;
			sum-=ret;//这个节点的流被消耗的 此时有可能sum==0 ,故for循环要多一个判断sum>0
		}	
	}
	return os-sum;//返回值必须是消耗的流
}
bool solve()
{
	int ans=0;
	while(bfs())
	{
		ans+=dinic(0,inf);
	}
	if(ans<n)
		return false;
	else
		return true;
}
bool func(int begin,int end)//每次调用都重新构图一次,蛋疼。
{
	k=1;
	memset(edge,0,sizeof(edge));
	memset(edgehead,0,sizeof(edgehead));
	for(int i=1;i<=n;i++)
	{
		addedge(0,i,1);
	}
	for(int i=1;i<=b;i++)
	{
		addedge(n+i,final,barn[i]);
	}
	for(int i=1;i<=n;i++)
		for(int j=begin;j<=end;j++)
		{
			addedge(i,mat[i][j]+n,1);
		}
	return solve();
}
int main()
{
	scanf("%d%d",&n,&b);
	final=n+b+1;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=b;j++)
			scanf("%d",&mat[i][j]);
	for(int i=1;i<=b;i++)
		scanf("%d",&barn[i]);
	int ans;
	int s=0,l=b;
	int mid;
	while(s<l)
	{
		mid=(s+l)/2;
		bool flag=false;
		for(int j=1;j+mid<=b;j++)
			if(func(j,j+mid))
			{
				flag=true;
				ans=mid;
				l=mid;
				break;
			}
		if(!flag)
		{
			s=mid+1;
		}
	}
	printf("%d\n",ans+1);
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值