正方形 dfs

蒜头君手上有一些小木棍，它们长短不一，蒜头君想用这些木棍拼出一个正方形，并且每根木棍都要用到。 例如，蒜头君手上有长度为 11，22，33，33, 33 的 55 根木棍，他可以让长度为11，22 的木棍组成一条边，另外三根分别组成 33 条边，拼成一个边长为 33 的正方形。蒜头君希望你提前告诉他能不能拼出来，免得白费功夫。 

输入格式

首先输入一个整数 n(4 \le n \le 20)n(4≤n≤20)，表示木棍数量，接下来输入 nn 根木棍的长度 p_i(1 \le p_i \le 10000)pi​(1≤pi​≤10000)。

输出格式

如果蒜头君能拼出正方形，输出"Yes"，否则输出"No"。

正确代码：

一条边成功凑出sum时进行下一条边的拼凑，拼完三条边即可返回。

#include<stdio.h>
int n,p[25];
int vis[25];
int sum,ok;
void dfs(int cnt,int s,int id) {
    if(ok) return;   
    if(s>sum) return;
    if(id==4) {ok=1;return;}   
    if(s==sum) {
        dfs(0,0,id+1);
    }
    for(int i=cnt;i<n;i++) {
        if(!vis[i]){
            vis[i]=1;
            dfs(i+1,s+p[i],id);
            vis[i]=0;
        }
    }
}
int main() {
    scanf("%d",&n);
    for(int i=0;i<n;i++) {
        scanf("%d",&p[i]);
        sum+=p[i];
    }
    sum=sum/4;
    ok=0;
    dfs(0,0,1);
    if(ok) printf("Yes");
    else printf("No");
    return 0;
}

 

超时版本：

每个木棍依次到每条边

#include<stdio.h>
int n,p[25];
int sum,ok;
void dfs(int id, int p1, int p2, int p3, int p4) {
    if(ok) return;   
    if(p1>sum||p2>sum||p3>sum||p4>sum) return;   
    if(id==n) {
        if(p1==sum&&p2==sum&&p3==sum&&p4==sum) {
            ok=1;printf("yes");exit(1);
        }
        return;
    }
    dfs(id+1,p1+p[id],p2,p3,p4);
    dfs(id+1,p1,p2+p[id],p3,p4);
    dfs(id+1,p1,p2,p3+p[id],p4);
    dfs(id+1,p1,p2,p3,p4+p[id]);
}
int main() {
    scanf("%d",&n);
    for(int i=0;i<n;i++) {
        scanf("%d",&p[i]);
        sum+=p[i];
    }
    sum=sum/4;
    ok=0;
    dfs(0,0,0,0,0);
    if(ok) printf("Yes");
    else printf("No");
    return 0;
}