20210427重温牛客比赛涂墙问题--dfs--剪枝--回溯

链接：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;
}