深搜剪枝之正方形

#include <cstdio>
#include <vector>

using namespace std;

const int kMax = 20 + 5;
const int kMaxEdge = 4;

int n, sum;
int p[kMax];
int len[kMaxEdge];
bool vis[kMax];

bool solve(int pos, int cur) {
    if(cur == kMaxEdge) return true;//搜到了三条边
    while(vis[pos] && pos < n) ++ pos;//找还没被用的木棍
    if(pos == n || len[cur] > sum / 4) return false;//这条边大于sum/4了，该剪枝。
    
    vis[pos] = true;//该木棍已经被用了
    len[cur] += p[pos];//拼接成边
    if(len[cur] == sum / 4 && solve(0, cur + 1)) return true;//当前边及剩下的边都等于sum/4，注意pos重置为0了。
    if(solve(pos + 1, cur)) return true;//接当前的棍子，然后再接下一根棍子
    len[cur] -= p[pos];//回溯，拆棍子
    vis[pos] = false;//把木棍拆下来了，取消标记
    if(solve(pos + 1, cur)) return true;//不接当前的棍子，接下一根棍子
    return false;//没搜到
}

int main() {
    scanf("%d", &n);
    for(int i = 0;i < n;++ i) {
        scanf("%d", &p[i]);
        sum += p[i];
    }
    if(sum % 4 == 0 && solve(0, 0)) printf("Yes\n");//正方形四条边相等
    else printf("No\n");
    return 0;
}