dfs 等边三角形 正方形

在计蒜客做到两道dfs的题目，两道题的题干基本一样，除了要创造出来的图形从等边三角形变成了正方形，但是解决的思路都不太一样。如果按照做等边三角形的做法去做正方形的话是会超时的。
注意，题干中说明，所有的木棍都需要用上

等边三角形

这个题的思路是，木棒要么放在第一条边，要么放在第二条边，要么第三条边，所以我们dfs是将情况分为这三种。
为了更快的结束，设置一个flag，当得到答案的时候，return

#include <iostream>
#include <stack>
#include <math.h>
using namespace std;
int n, k, flag = 0, maxx = 0, sum = 0;
long long M;
int edges[21];
int marked[21] = {0};
int group[3] = {0};


void dfs(int i, int group_id){
    group[group_id] += edges[i];
    if (group[0] > sum/3.0 || group[1] > sum/3.0 || group[2] > sum/3.0  ) {
        group[group_id] -= edges[i];
        return;
    }
    if (group[0] == sum / 3.0 || group[1] == sum / 3.0) {
        flag = 1;
        return;
    }
    if (i + 1 < n) {
        if(!flag) dfs(i + 1, 0);
        if(!flag) dfs(i + 1, 1);
        if(!flag) dfs(i + 1, 2);
    }
    group[group_id] -= edges[i];
    return;
}

int main()
{
    std::ios::sync_with_stdio(false);
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> edges[i];
        sum += edges[i];
    }
    dfs(0, 0);
    if (flag) {
        cout << "yes";
    }else{
        cout << "no";
    }
    return 0;
}  

正方形

这个题目的思想是，一条边一条边的搜。怎么个一条边一条边的搜呢，那就是，由于正方形边长为所有边长和的四分之一，我们就用这四分之一（设置为一个dfs的参数onel）减去合适的小木棒的数量，当onel变成零的时候，再将onel设为和的四分之一，直到找到合适的解。
来源：China震震

#include<iostream>
using namespace std;
int l[30];
bool mark[30];
int n,sum,one;
bool cmp(int a,int b)
{
    return a>b;
}
int flag=0;
void dfs(int onel,int num,int pos)
{
    if (flag) {
        return;
    }
    if (num == 3) {
        flag = 1;
        return;
    }
    if (onel == 0) {
        num++;
        onel = one;
        pos = 0;
    }
    for (int i = pos; i < n; i++) {
        if (!mark[i]) {
            if (onel >= l[i]) {
                mark[i] = true;
                dfs(onel - l[i], num, i);
                mark[i] = false;
            }
        }
    }
}

int main()
{
    fill(mark,mark+30,false);
    cin>>n;
    int sum=0;
    for(int i=0;i<n;i++){
        cin>>l[i];
        sum+=l[i];
    }
    one = sum/4;
    sort(l,l+n,cmp);
    if(l[0]>one||sum%4) cout<<"No"<<endl;
    else{
        dfs(one,0,0);
        if(flag==1) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
}