题目描述
«问题描述:
假设有n根柱子,现要按下述规则在这n根柱子中依次放入编号为1,2,3,…的球。
(1)每次只能在某根柱子的最上面放球。
(2)在同一根柱子中,任何2个相邻球的编号之和为完全平方数。
试设计一个算法,计算出在n根柱子上最多能放多少个球。例如,在4 根柱子上最多可放11 个球。
«编程任务:
对于给定的n,计算在n根柱子上最多能放多少个球。
输入格式
第1 行有1个正整数n,表示柱子数。
输出格式
程序运行结束时,将n 根柱子上最多能放的球数以及相应的放置方案输出。文件的第一行是球数。接下来的n行,每行是一根柱子上的球的编号。
题解:
按照完全平方数建图,枚举当前加的点,然后向残余网络种加边和加点,如果跑出来是有流量的,说明这个球被放在了已有的柱子上,否则一个增加一个柱子单独放。
因为每个球只能用一次,所以将球分成入点和出点,入点连接超级源点,出点连接超级汇点,至于完全平方数 j < i j<i j<i 且 i + j i+j 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;
}