洛谷P2765 魔术球问题

题目描述

«问题描述：

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

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

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

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

«编程任务：

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

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1： 

4

输出样例#1： 

11
1 8
2 7 9
3 6 10
4 5 11

说明

感谢 @PhoenixEclipse 提供spj

4<=n<=55

这道题是一个非常好网络流题目

我们先枚举球，一个个加到图里

并在加边的过程是三个操作：

当然不可缺少的是拆点把一个球x拆成x和x'

下一步x连源点，x'连汇点都是容量为一的有向边
第三步是找这个数可以与那个数组成完全平方数，假设这个数为y，则y向x'连边

下面就是跑最大流了，在跑最大流的时候记录一下这个点增广出来的点。

注：此题还可以用 匈牙利算法 做，但由于这题是 网络流24题 之一，于是用了 dinic 。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<cmath>
#include<cstring>
#define MAXN 10010
#define MAX 999999999
using namespace std;
int n,s,t,c=2;
int head[MAXN],deep[MAXN],real[MAXN],fa[MAXN<<1];
bool used[MAXN<<1];
struct node{
	int next,to,w;
}a[(MAXN/10*MAXN/10)<<1];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
inline void add(int u,int v,int w){
	a[c].to=v;a[c].w=w;
	a[c].next=head[u];
	head[u]=c++;
	a[c].to=u;a[c].w=0;
	a[c].next=head[v];
	head[v]=c++;
}
bool bfs(){
	int u,v;
	queue<int> q;
	for(int i=s;i<=t;i++)deep[i]=0;
	deep[s]=1;
	q.push(s);
	while(!q.empty()){
		u=q.front();
		q.pop();
		for(int i=head[u];i;i=a[i].next){
			v=a[i].to;
			if(a[i].w&&!deep[v]){
				deep[v]=deep[u]+1;
				if(v==t)return true;
				q.push(v);
			}
		}
	}
	return false;
}
int dfs(int x,int limit){
	if(x==t)return limit;
	int v,sum,cost=0;
	for(int i=head[x];i;i=a[i].next){
		v=a[i].to;
		if(a[i].w&&deep[v]==deep[x]+1){
			sum=dfs(v,min(a[i].w,limit-cost));
			if(sum>0){
				a[i].w-=sum;
				a[i^1].w+=sum;
				cost+=sum;
				fa[x>>1]=v>>1;
				if(cost==limit)break;
			}
			else deep[v]=-1;
		}
	}
	return cost;
}
int dinic(){
	int ans=0;
	while(bfs())ans+=dfs(s,MAX);
	return ans;
}
int main(){
	int now=0,num=0;
	n=read();
	s=0;t=MAXN-10;
	memset(used,false,sizeof(used));
	while(num<=n){
		now++;
		add(s,now<<1,1);
		add(now<<1|1,t,1);
		for(int i=sqrt(now)+1;i*i<(now<<1);i++)add((i*i-now)<<1,now<<1|1,1);
		if(!dinic())real[++num]=now;
	}
	printf("%d\n",now-1);
	for(int i=1;i<=n;i++)
	if(!used[real[i]]){
		for(int j=real[i];j!=(t>>1)&&j!=0;j=fa[j]){
			printf("%d ",j);
			used[j]=true;
		}
		printf("\n");
	}
	return 0;
}