11H:迷宫入口（dfs）

总时间限制: 1000ms

内存限制: 10000kB

描述

爱好探险的你，找到一座装满了宝藏的迷宫的入口，你看到入口的大门上有一个边长为s的正方形的大锁，旁边散落着n块正方形的金属小片，你意识到锁的钥匙，即是用这n小块，拼成大门上的正方形，你想知道是否能拼成这把钥匙打开迷宫的大门。

输入

输入包含多组数组，第一行是一个整数t（1 <= t <= 10)，表示有t组数据。接下里每组数组包含整数s，即锁的边长，整数n（1 <= n <= 16)，即金属小片的数量，接下来n个整数，分别是各个小片的边长ci（1 <= ci <= 10）。

输出

每组数据输出一行，输出“YES"或者"NO"，表示是否可以打开大门。

样例输入

2
4 8 1 1 1 1 1 3 1 1
5 6 3 3 2 1 1 1

样例输出

YES
NO

/*
题型：dfs
难点：固定什么，改变什么
不同的点，可以放不同的正方形
那到底是搜索正方形放在哪一个位置呢，还是搜索一个位置放哪个正方形呢？
显然前者的盲目性太强，会有很多重复情况（就算一上来你先枚举边长最大的正方形，那也可以放好多地方）
对于后者，本题的边长只有10种情况，所以状态空间会少很多
*/
#include <iostream>
#include <algorithm>
#include <cstring>
#include <climits>
#include <cmath>
#include <numeric>
#include <vector>
using namespace std;
#define _for(i,a,b) for(int i = (a); i < (b); ++i)
#define _rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define _zero(x) memset(x, 0, sizeof(x))
#define _init(x, v) memset(x, v, sizeof(x))

int s, n;
bool b[45][45];
int c[15]; // c[l]记录长度为l的小正方形有多少个

void in(){
    _zero(b);
    _zero(c);
    cin >> s >> n;
    for(int i = 0; i < n; ++i){
        int t;
        cin >> t;
        ++c[t];
    }
}

bool canPut(int i, int j, int d){
    if(i+d-1 > s || j+d-1 > s) return 0;
    for(int ii = 0; ii < d; ++ii)
        for(int jj = 0; jj < d; ++jj)
            if(b[i+ii][j+jj])
                return 0;
    return 1;
}
void put(int i, int j, int d, int v){
    for(int ii = 0; ii < d; ++ii)
        for(int jj = 0; jj < d; ++jj)
            b[i+ii][j+jj] = v;
}

bool dfs(int dep){ // 深度dep虽然没有用到，但是便于打印调试
    int i = 1, j = 1;
    for(; i <= s; ++i){
        for(j = 1; j <= s; ++j){
            if(b[i][j] == 0)
                goto END;
        }
    }
    END:;
    if(i == s+1 && j == s+1) return 1; // 跑到右下角的右下方了，说明空位都填完了，成功
    for(int l = 1; l <= 10; ++l){
        if(c[l] && canPut(i, j, l)){
            c[l]--;
            put(i, j, l, 1);
            if(dfs(dep+1)) return 1;
            put(i, j, l, 0);
            c[l]++;
        }
    }
    return 0;
}

int main(){
    int T;
    cin >> T;
    while(T--){
        in();
        bool ans = dfs(0);
        if(ans) cout << "YES\n";
        else cout << "NO\n";
    }

    return 0;
}