題目描述 «問題描述：

假設有n根柱子，現要按下述規則在這n根柱子中依次放入編號為1，2，3，…的球。

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

（2）在同一根柱子中，任何2個相鄰球的編號之和為完全平方數。

試設計一個演算法，計算出在n根柱子上最多能放多少個球。例如，在4 根柱子上最多可放11 個球。

«程式設計任務：

對於給定的n，計算在n根柱子上最多能放多少個球。

輸入輸出格式 輸入格式： 第1 行有1個正整數n，表示柱子數。

輸出格式： 程式執行結束時，將n 根柱子上最多能放的球數以及相應的放置方案輸出。檔案的第一行是球數。接下來的n行，每行是一根柱子上的球的編號。

[分析] 網路流的解法網路上已經，很多了。這裡講一下貪心的解法。 貪心的解法，比網路流的好理解很多。特別簡單 就是如果當前球能放在使用過的杆子上就放使用過的杆子上，如果不能再放沒放過的。 並且，放球后在他下面的那個球必定比它小，所以不存在多種情況的分支（思考一下）。 所以模擬一邊就好了。

[程式碼]

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<vector>
#include<cmath>
#include<cstring>
using namespace std;

vector<int>v[57];

bool judge(int x)
{
	int a = sqrt(x);
	if (a*a == x)return 1;
	else return 0;
}

int main()
{

	
	int n;
	while (scanf("%d", &n) != EOF)
	{
		memset(v, 0, sizeof(v));
		int cnt = 0;
		int num = 0;
		while (cnt <= n)
		{
			int flag = 0;
			++num;
			for (int i = 1; i <= cnt; i++)
			{
				if (judge(v[i][v[i].size() - 1] + num))
				{
					flag = 1;
					v[i].push_back(num);
					break;
				}
			}
			if (flag == 0)
			{
				cnt++;
				if (cnt > n)break;
				v[cnt].push_back(num);
			}
		}
		printf("%d\n", num-1);
		for (int i = 1; i <= cnt; i++)
		{
			for (int j = 0; j < v[i].size(); j++)
			{
				if (j)printf(" ");
				printf("%d", v[i][j]);
			}
			printf("\n");
		}
	}
}