魔术球问题
题目描述
假设有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;
}