洛谷P2765：魔术球问题

(这题很舒服。。。)

题目大意

      题目写得很清楚。 

      假设有n根柱子，现要按下述规则在这n根柱子中依次放入编号为1，2，3，...的球。

      （1）每次只能在某根柱子的最上面放球。

      （2）在同一根柱子中，任何2个相邻球的编号之和为完全平方数。

       试设计一个算法，计算出在n根柱子上最多能放多少个球。例如，在4 根柱子上最多可放11 个球。

      «编程任务：

      对于给定的n，计算在n根柱子上最多能放多少个球。

解法

      没错，就是这样，是不是马上想到了暴力dfs！！！n只有55，但是你无法确定球的个数，所以dfs暴枚肯定会很慢（wjh：不然怎么出在网络流24题里）所以，我们要往网络流方面去想，而我们又要运用到网络流的反悔性质，所以很明显就会想到拆点.

      我们每一次把一个球拆成两个点，前面的点连（源点），后面的点连（汇点），权值都为1，然后中间的怎么连呢？

      很明显我们回记起来还有一个条件没有用上，那就是同一个柱子中，两个相邻的球的编号之和为完全平方数，所以如果a+b为完全平方数，那么我们就把a的前面那个店和b后面那个点连接起来，权值为一（注意：这里可能会理解错误，因为有些人可能想问x后面可以跟y，y后面可以跟z，那么是不是要把y的后面那个点与y前面那个点连起来，这样跑出来的结果才会对。。错！！！因为这个网络流中的流量只有两种状态，无解和有解，无解时跑出来的结果为0，有解时跑出来的结果>0，这样才能实现一次装一个柱子，要不你一次装多个柱子，那你怎么加点。。对吧!？）

      我们每次加一个点，然后跑一次网络流，如果有解，那么就继续加点，否则，就将它压在第二个柱子的最底下，也就是里面的head[]数组。

说完了，看代码，喜欢我的点一波关注哦！

代码

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>

int n;//共有柱子 
int spend=0;//开始用第几根柱子 
int tot_ball=0;
int begin,end;
int head[60];//第i根柱子在柱底的是第几号球 
struct edge{int y,next,c;};
edge s[400010];
int first[200010];
int match[50010];
int len=1;
int h[50010];
int f[100010];

void ins(int x,int y,int c)
{
	len++;
	s[len].y=y;s[len].c=c;
	s[len].next=first[x];first[x]=len;
}

int min(int x,int y)
{
	return x<y?x:y;
}

bool bfs()
{
	int st=1,ed=2;
	f[st]=begin;
	memset(h,0,sizeof(h));
	h[begin]=1;
	while(st!=ed)
	{
		int x=f[st];
		for(int i=first[x];i!=0;i=s[i].next)
		{
			int y=s[i].y;
			if(h[y]==0 && s[i].c>0)
			{
				h[y]=h[x]+1;
				f[ed]=y;
				ed++;
				if(ed==100000) ed=1;
			}
		}
		st++;
		if(st==100000) st=1;
	}
	return h[end];
}

int dfs(int x,int t)
{
	if(x==end) return t;
	int tot=0,flow=0;
	for(int i=first[x];i!=0;i=s[i].next)
	{
		if(tot==t) return t;
		int y=s[i].y;
		if(s[i].c>0 && h[y]==h[x]+1)
		{
			flow=dfs(y,min(s[i].c,t-tot));
			tot+=flow;s[i].c-=flow;s[i^1].c+=flow;
			if(flow!=0 && y!=begin && y!=end && x%2==0)
				match[x>>1]=y>>1;
		}
	}
	if(tot==0) h[x]=0;
	return tot;
}

int dinic()
{
	int tot=0;
	int dx;
	while(bfs())
	{
		dx=dfs(begin,1e9);
		while(dx!=0)
		{
			tot+=dx;
			dx=dfs(begin,1e9);
		}
	}
	return tot;
}

int main()
{
	scanf("%d",&n);
	begin=1;end=50000;
	while(spend<=n)
	{
		tot_ball++;
		ins(begin,tot_ball<<1,1);ins(tot_ball<<1,begin,0);
		ins((tot_ball<<1)|1,end,1);ins(end,(tot_ball<<1)|1,0);
		for(int i=sqrt(tot_ball)+1;i*i<2*tot_ball;i++)
		{
			ins((i*i-tot_ball)<<1,(tot_ball<<1)|1,1),ins((tot_ball<<1)|1,(i*i-tot_ball)<<1,0);
		}
		int k=dinic();
		if(k==0) //因为没有流量流到汇点（也就是说，这个球只能放到下一个柱子了）； 
		{//所以我们把它设为下一个柱子的开头 
			spend++;
			head[spend]=tot_ball;
		}
	}
	printf("%d\n",tot_ball-1);
	bool visit[50010];//第i号球是否用过 
	memset(visit,false,sizeof(visit));
	for(int i=1;i<=n;i++)
	{
		if(visit[head[i]]) continue;
		int now=head[i];
		while(now!=0 && visit[now]==false)
		{
			printf("%d ",now);
			visit[now]=true;
			now=match[now];
		}
		printf("\n");
	}
}