洛谷 P2765 魔术球问题

题目描述

«问题描述:

假设有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;
}

 

posted @ 2017-07-20 08:00  wawcac  阅读(171)  评论(0编辑  收藏  举报