7.4 魔术球问题弱化版-CSDN博客

题目描述

假设有 n 根柱子，现要按下述规则在这 n 根柱子中依次放入编号为 1，2，3，…的球。

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

（2）在同一根柱子中，任何 2 个相邻球的编号之和为完全平方数。

试设计一个算法，计算出在 n 根柱子上最多能放多少个球。例如，在 4 根柱子上最多可放 11 个

对于给定的 n，计算在 n 根柱子上最多能放多少个球。

输入描述

第 1 行有 1 个正整数 n，表示柱子数。

输出描述

一行表示可以放的最大球数

样例输出。

样例输入

题目限制（为什么说弱化版就在这里）

N<=60，时限为3s；比起原题还有弱化在不用打出方案，方案太坑了

本题为一个网络流问题的弱化版，结果弱化成模拟了。。。

所以，就直接按模拟做吧，用了一点小技巧：

1.贪心，尽可能多地让球都叠加在一个柱子上，毕竟，即使一个数没有能与它组成平方数，他还可以独占一个柱子

贪心的正确性证明：

能够证明，按照贪心的思路，每一步得到的都是最优解，首先，只有当出现一个数必须另开一列放的时候，才能会出现最优解的问题。

如果当前的数a必须单开一列，那么所有能与a组成平方数的数字b，b上面都已经放了数字，如果将a,b调换，那么b将无处可去，也只能单开一列，每一列的最上层数字也没有改变，所以对于之后所有的数字，他们的相对位置（放在哪个球上面/单开一列）没有任何变化，解不会变差。

2.既然只是相邻组成平方数，那就没必要考虑下面的球的编号，只需储存最上面球的编号即可判断某个球能否放在这根柱子上

#include<cstdio>
int n,i,t=1,book=1;
int s[105];
int pd(int x)
{
　　for(int j=1;j*j<=x;j++)
　　{
　　　　if(j*j==x) return 1;
　　}
　　return 0;
}
int main()
{
　　scanf("%d",&n);
　　while(book==1)
　　{
　　　　book=0;
　　　　for(i=1;i<=n;i++)
　　　　{
　　　　　if(pd(t+s[i])==1)
　　　　　　{
　　　　　　　　s[i]=t;
　　　　　　　　t++;
　　　　　　　　book=1;
　　　　　　　　break; 
　　　　　　}
　　　　　　if(s[i]==0)
　　　　　　{
　　　　　　　　s[i]=t;
　　　　　　　　t++;
　　　　　　　　book=1;
　　　　　　　　break;
　　　　　　}
　　　　}
　　}
　　printf("%d",t-1);
　　return 0;
}