bzoj 4405: [wc2016]挑战NPC【带花树】

把每个筐子拆成3个,分别表示放0/1/2个,然后把这三个点两两连起来,每一个可以放在筐里的球都想这三个点连边。
这样可以发现,放0个球的时候,匹配数为1,放1个球的时候,匹配数为1,放2个球的时候,匹配数为2,放3个球的时候,匹配数为3。
所以直接把带花树的匹配数-n即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=1005;
int T,n,m,k,h[N],cnt,c[N],lk[N],v[N],ti,ans,pr[N],f[N];
queue<int>q;
struct qwe
{
	int ne,to;
}e[N*N];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
void add(int u,int v)
{
	cnt++;
	e[cnt].ne=h[u];
	e[cnt].to=v;
	h[u]=cnt;
}
void ins(int x,int y)
{
	add(x,y);
	add(y,x);
}
int zhao(int x)
{
	return x==f[x]?x:f[x]=zhao(f[x]);
}
int lca(int x,int y)
{
	for(ti++;;swap(x,y))
		if(x)
		{
			x=zhao(x);
			if(v[x]==ti)
				return x;
			v[x]=ti;
			x=lk[x]?pr[lk[x]]:0;
		}
}
void hb(int x,int y,int k)
{
	while(zhao(x)!=k)
	{
		pr[x]=y;
		int z=lk[x];
		if(c[z]==1)
		{
			c[z]=0;
			q.push(z);
		}
		if(zhao(x)==x)
			f[x]=k;
		if(zhao(z)==z)
			f[z]=k;
		y=z;
		x=pr[y];
	}
}
bool bfs(int fr)
{
	for(int i=1;i<=n+m*3;i++)
		c[i]=-1,f[i]=i;
	while(!q.empty())
		q.pop();
	q.push(fr);
	c[fr]=0;
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=h[u];i;i=e[i].ne)
		{
			if(c[e[i].to]==-1)
			{
				pr[e[i].to]=u;
				c[e[i].to]=1;
				if(!lk[e[i].to])
				{
					for(int tmp,la,now=e[i].to;now;now=la)
					{
						la=lk[tmp=pr[now]];//!!!!!
						lk[now]=tmp;
						lk[tmp]=now;
					}
					return 1;
				}
				c[lk[e[i].to]]=0;
				q.push(lk[e[i].to]);
			}
			else if(c[e[i].to]==0&&zhao(u)!=zhao(e[i].to))
			{
				int lc=lca(u,e[i].to);
				hb(u,e[i].to,lc);
				hb(e[i].to,u,lc);
			}
		}
	}
	return 0;
}
int main()
{
	T=read();
	while(T--)
	{
		memset(h,0,sizeof(h));
		memset(c,0,sizeof(c));
		memset(lk,0,sizeof(lk));
		memset(v,0,sizeof(v));
		cnt=0,ans=0,ti=0;
		n=read(),m=read(),k=read();
		for(int i=1;i<=m;i++)
		{
			ins(i,i+m);
			ins(i+m,i+m*2);
			ins(i+m*2,i);
		}
		while(k--)
		{
			int x=read(),y=read();
			ins(x+3*m,y);
			ins(x+3*m,y+m);
			ins(x+3*m,y+m*2);
		}
		for(int i=1;i<=3*m+n;i++)
			if(!lk[i])
				ans+=bfs(i);
		printf("%d\n",ans-n);
	}
	return 0;
}
posted @ 2018-03-15 11:00  lokiii  阅读(118)  评论(0编辑  收藏  举报