题意:给你n 个盘,按照要求摆放球:球只能从1,2,3,........编号开始,相邻的两个球的编号和必须是平方数,
问最多能摆放多少个球。
省赛后的第一道水题(不知道为什么这么少人做),dfs 水过,每次从第一个盘开始放,如何可以放到当前盘,则更新它的球号,直到不能放。
#include<cstdio>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<map>
#include<cmath>
#include<iostream>
#include <queue>
#include <stack>
#include<algorithm>
#include<set>
using namespace std;
#define INF 1e8
#define inf -0x3f3f3f3f
#define eps 1e-8
#define ll __int64
#define maxn 250001
#define mol 100007
int dp[55];
int n,ans;
void dfs(int m,int d)
{
if(d>n)
{
if(m>ans)
ans=m-1;
return ;
}
int tmp;
double cnt;
cnt=sqrt((double)(dp[d]+m));
tmp=(int)cnt;
if(cnt-tmp<eps||dp[d]==0)
{
dp[d]=m;
m++;
d=1;
dfs(m,d);
m--;
}
else dfs(m,d+1);
}
int main()
{
int t,i,j;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(dp,0,sizeof(dp));
ans=0;
dfs(1,1);
printf("%d\n",ans);
}
return 0;
}