洛谷2765 魔術球問題

題目描述

«問題描述：

假設有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個柱子

#include<cmath>
#include<cstdio>
using namespace std;
int a[3001][3001];
int main()
{
    int n;
    scanf("%d",&n);
    int now=1,sum=0;
    bool ok; int tmp;
    while(1)
    {
        ok=false;
        for(int i=1;i<=sum;i++)
        {
            if((int)sqrt(now+a[i][a[i][0]])==sqrt(now+a[i][a[i][0]]))
            {
                ok=true;
                a[i][++a[i][0]]=now;
                break;
            }
        } 
        if(!ok) 
        {
            if(sum<n)a[++sum][++a[sum][0]]=now; 
            else break;
        }
        now++;
    }
    printf("%d\n",now-1);
    for(int i=1;i<=sum;i++)
    {
        for(int j=1;j<=a[i][0];j++) printf("%d ",a[i][j]);
        printf("\n");
    }
}

判完全平方數無限WA：

if(sqrt(1.0*tmp)*sqrt(1.0*tmp)==1.0*tmp)