Square

Square

dfs主要是注意回溯时候的剪枝操作

#include<bits/stdc++.h>
using namespace std;
int n;
int a[25];
int f[25];
int goal = 0;
int m;
int flag = false;

void dfs(int n, int last, int now) {
    if(n == 5) {
        flag = true;
        return ;
    }
    if(now == goal) {
        int i;
        for(i = m; i >= 1; i--) {
            if(f[i] == 0) {
                f[i] = 1;
                break;
            }
        }
        dfs(n + 1, i, a[i]);
        if(flag) return;
        //表示已经找到答案 直接return
        f[i] = 0;
    }
    for(int i = last - 1; i >= 1; i--) {
        if(now + a[i] <= goal && f[i] == 0) {
            f[i] = 1;
            dfs(n, i, now + a[i]);
            if(m - i + 1 == n - 1) {
                return ;
                //如果当前最大边还没有被选中那么这种拼接方式并不可以构成正方形，return 掉
            }
            if(flag) {
                break;
                 //表示已经找到答案 直接break掉
            }
            f[i] = 0;
            while(a[i - 1] == a[i]) {
            //把长度一样的边掠过
                i--;
            }
        }

    }

}

int main() {
    scanf("%d", &n);
    while(n--) {
        scanf("%d", &m);
        for(int i = 1; i <= m; i++) {
            f[i] = 0;
        }
        flag = false;
        int sum = 0;
        int l = 0;
        for(int i = 1; i <= m; i++) {
            scanf("%d", &a[i]);
            sum += a[i];
            l = max(l, a[i]);
        }
        if(sum % 4 != 0 || l > sum / 4) {
            pr匈牙利intf("no\n");
            continue;
        }
        goal = sum / 4;
        sort(a + 1, a + 1 + m);
        dfs(1, m, 0);
        if(flag) {
            printf("yes\n");
        }else {
            printf("no\n");
        }
    }
    return 0;
}