魔术球问题

假设有n根柱子，现要按下述规则在这n根柱子中依次放入编号为1，2，3，…的球。
（1）每次只能在某根柱子的最上面放球。
（2）在同一根柱子中，任何2个相邻球的编号之和为完全平方数。
试设计一个算法，计算出在n根柱子上最多能放多少个球。例如，在4 根柱子上最多可放11 个球。
«编程任务：
对于给定的n，计算在n根柱子上最多能放多少个球。
输入格式：
第1 行有1个正整数n，表示柱子数。
输出格式：
程序运行结束时，将n 根柱子上最多能放的球数以及相应的放置方案输出。文件的第一行是球数。接下来的n行，每行是一根柱子上的球的编号。

我们要装最多的球——装N个球用最多N个柱子，最少用几个柱子——最小线段覆盖。
可以枚举加边，可以二分答案。

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

const int MAXN=1e6;
const int INF=1e9;
const int base=5e5;
int n,s,t,num=0,ans=0;

struct edge{
    int to,next,w;
}e[MAXN<<2];
int head[MAXN],cnt=1;

inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}

int fa[MAXN],rk[MAXN];

void makeset(int n){fa[n]=n;}
int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
void unset(int x,int y){
    if((x=find(x))==(y=find(y)))return;
    if(rk[x]>rk[y])fa[y]=x;
    else {
        fa[x]=y;
        if(rk[x]==rk[y])rk[y]++;
    }
}

int dep[MAXN],cur[MAXN];
bool po[MAXN],used[2000][2000];
queue<int>q;
bool bfs(int x){
    memset(dep,0,sizeof(dep));
    q.push(x);dep[x]=1;
    while(q.size()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].to,w=e[i].w;
            if(w&&!dep[v]){
                dep[v]=dep[u]+1;q.push(v);
            }
        }
    }
    if(!dep[t])return 0;
    return 1;
}

int dfs(int u,int flow){
    if(u==t||flow==0)return flow;
    for(int &i=cur[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(w&&dep[v]==dep[u]+1){
            int tem=dfs(v,min(flow,w));
            if(tem){
                if(v!=t&&u!=s){
                //  cout<<u<<":"<<v-base<<endl;
                    unset(u,v-base);    
                }
                e[i].w-=tem;
                e[i^1].w+=tem;
                return tem;
            }
        }
    }
    return 0;
}

int dinic(int n){
    makeset(n);
    int tem=ans;
    while(bfs(s)){
        for(int i=0;i<=n;i++)cur[i]=head[i];
        for(int i=base+1;i<=base+n;i++)cur[i]=head[i];
        cur[t]=head[t];
        while(int d=dfs(s,INF))tem+=d; 
    }
    ans=tem;
    return n-tem;
}

bool vis[MAXN];

int main(){
//  freopen("1.in","r",stdin);
//  freopen("1.out","w",stdout);
    memset(po,0,sizeof(po));
    memset(used,0,sizeof(used));
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=500;i++)po[i*i]=1;
    scanf("%d",&n);
    s=0;t=MAXN-5;
    while(1){
        ++num;
        //memset(e,0,sizeof(e));
        add(s,num,1);add(num,s,0);
        add(base+num,t,1);add(t,base+num,0);
    /*  for(int i=1;i<=num;i++){//出 
            add(s,i,1);add(i,s,0);
        }
        for(int i=num+1;i<=(num<<1);i++){//入 
            add(i,t,1);add(t,i,0);
        }*/
        for(int i=num-1;i>=1;i--){
            if(po[num+i]&&!used[num][i]){
                add(i,num+base,1);add(num+base,i,0);
                used[num][i]=1;
                //cout<<i<<" "<<i+num<<endl; 
            }
        }
        int tmp=dinic(num);
        if(tmp>n){
            printf("%d\n",num-1);
            break;  
        }       
    }
    for(int i=1;i<=num-1;i++){
        int ly=fa[i];
        if(!vis[ly]){
            printf("%d ",i);vis[ly]=1;
            for(int j=i+1;j<=num-1;j++){
                if(fa[j]==ly)printf("%d ",j);
            }
            printf("\n");
        }
    }
    return 0;
}