链接:https://ac.nowcoder.com/acm/problem/220347
来源:牛客网
题目描述
母牛哥有一桶油漆,把它用完可以给n平方米的墙涂上颜色.
母牛哥想要在墙上涂5个正方形(这些正方形的边长都是整数,单位是米),并且刚好把油漆用完.
母牛哥能做到吗?
输入描述:
第一行一个数字t(<=1000),表示测试样例数量
接下来t行,每行一个数字n(0<=n<=1000000),表示母牛哥的油漆可以涂多少平方米.
输出描述:
输出t行,对于每个输入.
如果母牛哥能够做到,就输出YES.
否则输出NO.
示例1
输入
复制
2
4
55
输出
复制
NO
YES
说明
4显然不能分解成5个正平方数,所以这桶油漆不能涂5个正方形.
55可以涂5个正方形,他们面积分别是1 4 9 16 25.
题目:思路简单,dfs的简单题,但是比较麻烦的是剪枝条件,如果达不到最优剪枝会超时。
本题有两种解题方法:
这是dfs剪枝过程,只有dfs函数。
主函数中间有个比较恶心的操作就是初始化tree数组。
int main()
{
int t;
cin>>t;
while(t--){
//int n;
cin>>n;
for(int i=1000;i>=1;i--){
tree[i]=i*i;
}
f=0;
dfs(0,0);
if(f==0) cout<<"NO"<<endl;
else cout<<"YES"<<endl;
}
return 0;
}
因为数值是1000^2非常大,如果对其进行硬性初始化难免会超时,所以对其进行1000*1000操作,只要遍历1000个数,优化算法而且只要用到这么大的数,非常方便。
本题还有一个快捷方法,那就是直接枚举暴力,
也就是当你发现规律之后,发现就那么几个数不满足题意,所以枚举那些值,其他的都满足题意就可以了。
代码:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <string>
#include <set>
#include <map>
#define swap(a, b) { int temp = a; a = b; b = temp; }
typedef long long ll;
using namespace std;
int main(int argc, char* argv[])
{
int t;
cin >> t;
while (t--) {
int n;
cin >> n;
if (n==5)cout << "YES" << endl;
else if((n<=7)||n==9||n==10||n==12||n==15||n==18||n==33)cout << "NO" << endl;
else cout << "YES" << endl;
}
return 0;
}