[网络流24题]魔术球问题

魔术球问题

题目描述

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

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

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

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

编程任务：

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

输入输出格式

输入格式：

第1 行有1个正整数n，表示柱子数。

输出格式：

程序运行结束时，将n 根柱子上最多能放的球数以及相应的放置方案输出。文件的第一行是球数。接下来的n行，每行是一根柱子上的球的编号。

输入输出样例

输入样例#1：

4

输出样例#1：

111 82 7 93 6 104 5 11

题解;

枚举放的个数，将能放在上面的球连边，若流量相对上一个没有变化则说明需要加一个柱子，加到刚好小于就可以

代码：

#include<bits/stdc++.h>
using namespace std;

const int max_n = 3300;
const int max_m = 1000010;
const int inf   = 1e9+7;

int point[max_n],nxt[max_m],v[max_m],val[max_m],remain[max_m],remain1[max_m];
int point2[max_n],nxt2[max_m],v2[max_m],val2[max_m],remain2[max_m];
bool b[max_n];
int res[61][max_n],deep[max_n];
int n,tot,ans,num,m=1610,s,t,l;

inline void init()
{
	memset(point,-1,sizeof(point));
	memset(nxt,-1,sizeof(nxt));
	tot=-1;
	for(int i=1; i<=60; ++i)
	  b[i*i]=true;
	s=0; t=3290;
}

inline void addedge(int x,int y,int cal)
{
	++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; remain[tot]=cal; remain1[tot]=cal;
	++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; remain[tot]=0;   remain1[tot]=0;
} 

inline bool bfs(int s,int t)
{
	memset(deep,0x7f,sizeof(deep));
	queue<int> q;
	q.push(s); deep[s]=0;
	while(!q.empty())
	{
		int now=q.front(); q.pop();
		for(int i=point[now]; i!=-1; i=nxt[i])
		  if(deep[v[i]]>inf && remain[i])
		  {
		  	deep[v[i]]=deep[now]+1;
		  	q.push(v[i]);
		  }
	}
	return deep[t]<inf;
}

int dfs(int now,int t,int limit)
{
	if(now==t || !limit) return limit;
	int f,flow=0;
	for(int i=point[now]; i!=-1; i=nxt[i])
	  if(deep[v[i]]==deep[now]+1 && (f=dfs(v[i],t,min(limit,remain[i]))))
	  {
	  	flow+=f;
	  	limit-=f;
	  	remain[i]-=f;
	  	remain[i^1]+=f;
	  	if(!limit) break;
	  }
	return flow;
}

inline int dinic(int s,int t)
{
	int ans=0;
	while(bfs(s,t))
	  ans+=dfs(s,t,inf);
	return ans;
}

int main()
{
//	freopen("ballaplus.in","r",stdin);
//	freopen("ballaplus.out","w",stdout);
	scanf("%d",&n);
	init();
	int now=0,last=0;
	for(int k=1; k<=m; ++k)
	{
		l=tot;
		for(int i=0; i<=k; ++i) point2[i]=point[i];
		for(int i=0; i<=tot; ++i) nxt2[i]=nxt[i],v2[i]=v[i],remain2[i]=remain[i];
	    for(int i=0; i<=tot; ++i) remain[i]=remain1[i];
		for(int i=1; i<k; ++i)
		  if(b[i+k])
		  {
		  	addedge(i,k+m,1);
		    
		  } 
		addedge(s,k,1);
		addedge(k+m,t,1);
		now=dinic(s,t);
		if(now==last) num++;
		last=now;
		if(num>n)
		{
			ans=k;
			break;
		}  
	}
	printf("%d\n",--ans);
	for(int i=0; i<=ans; ++i) point[i]=point2[i];
	for(int i=0; i<=l; ++i) nxt[i]=nxt2[i],v[i]=v2[i],remain[i]=remain2[i];
	dinic(s,t);
	memset(b,0,sizeof(b));
	for(int i=1; i<=n; ++i)
	{
		int wh;
		for(int j=1; j<=ans; ++j)
		  if(!b[j]){wh=j; break;}
		while(wh!=0)
		{
			printf("%d ",wh);
			b[wh]=true;
			int now=wh;
			for(int j=point[now]; j!=-1; j=nxt[j])
			  if(!b[v[j]] && !remain[j] && v[j]!=s) wh=v[j]-m;
			if(wh==now) wh=0;
		} 
		printf("\n");
	}
	return 0;
}