题目描述
«问题描述：
假设有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
【题目大意】
n个柱子 往上放编号1--n的球 要求每根柱子上相邻球的编号和为完全平方数。
【思路】
贪心
能放就放 不能放放下一个柱子
【code】
#include<iostream> #include<cstdio> #include<cmath> using namespace std; int now=1,n,flag,num; int a[2000][2000]; int check(int x) { if(sqrt(x)*sqrt(x)==x)return 1; return 0; } int main() { scanf("%d",&n); while(1) { flag=0; for(int i=1; i<=num; i++) { if((int)sqrt(now+a[i][a[i][0]])==sqrt(now+a[i][a[i][0]])) { a[i][++a[i][0]]=now; flag=1; break; } } if(!flag) { if(num<n) a[++num][++a[num][0]]=now; else break; } now++; } printf("%d ",now-1); for(int i=1; i<=num; i++) { for(int j=1; j<=a[i][0]; j++) { printf("%d ",a[i][j]); } printf(" "); } return 0; }