P2765 魔术球问题

题目描述

«问题描述：

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

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

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

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

«编程任务：

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

输入格式

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

输出格式

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

---

题解：
按照完全平方数建图，枚举当前加的点，然后向残余网络种加边和加点，如果跑出来是有流量的，说明这个球被放在了已有的柱子上，否则一个增加一个柱子单独放。

因为每个球只能用一次，所以将球分成入点和出点，入点连接超级源点，出点连接超级汇点，至于完全平方数$j<i$ 且 $i+j$为完全平方数就让i的入点连接j的出点即可

---

AC代码：

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
#define LL long long
const int MAXN = 1e5+50;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int n,base=5000,s,t,st,tot=1,ans;
int head[MAXN],to[MAXN],w[MAXN],nxt[MAXN],h[MAXN],v[MAXN],vis[MAXN];
inline void ade(int u,int v,int ww){
    to[++tot]=v; w[tot]=ww; nxt[tot]=head[u]; head[u]=tot;
}
inline void add(int u,int v,int w){ ade(u,v,w); ade(v,u,0); }
inline int bfs(){
    queue<int> que; que.push(s); memset(h,0,sizeof(h)); h[s]=1;
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=head[u];i;i=nxt[i]){
            if(w[i] && !h[to[i]]){
                h[to[i]]=h[u]+1; que.push(to[i]);
            }
        }
    }
    return h[t];
}
inline int dfs(int x,int f){
    if(x==t) return f; int fl=0;
    for(int i=head[x];i&&f;i=nxt[i]){
        if(w[i] && h[to[i]]==h[x]+1){
            int mi=dfs(to[i],min(f,w[i]));
            w[i]-=mi; w[i^1]+=mi; fl+=mi; f-=mi;
        }
    }
    if(!fl) h[x]=-1;
    return fl;
}
inline int dinic(){
    int res=0;
    while(bfs()) res+=dfs(s,INF);
    return res;
}
inline void solve(){
    for(int i=1;i<st;i++){
        for(int j=head[i];j;j=nxt[j]){
            if(w[j]) continue;
            v[i]=to[j]-base; break;
        }
    }
    for(int i=1;i<st;i++){
        if(vis[i]) continue;
        int t=i;
        while(t!=-base){
            printf("%d ",t); vis[t]=1; t=v[t];
        } puts("");
    }
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    scanf("%d",&n); s=0,t=10000;
    while(true){
        ans++,st++;
        for(int i=1;i<st;i++)
            if(sqrt(i+st)==(int)sqrt(i+st))
                add(i,st+base,1);
        add(s,st,1); add(st+base,t,1);
        ans-=dinic();
        if(ans>n) break;
    }
    printf("%d\n",st-1);
    solve();
    return 0;
}