题目描述
«问题描述：
假设有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
解题思路
贪心明明可以，非要搞网络流……http://hzwer.com/1878.html
源代码
#include<stdio.h> #include<math.h> int n,i; int a[100][100]={0},num=0; inline bool judge(int x) { int s=sqrt(x); return s*s==x; } int main() { scanf("%d",&n); for(i=1;;i++) { bool flag=1; for(int pos=1;pos<=num;pos++) { if(judge(i+a[pos][a[pos][0]])) { a[pos][++a[pos][0]]=i; flag=0; break; } } if(flag) a[++num][1]=i,a[num][0]=1; if(num>n) break; } printf("%d\n",i-1); for(int i=1;i<=n;i++) { for(int j=1;j<=a[i][0];j++) printf("%d ",a[i][j]); printf("\n"); } return 0; }
