HDU1518 DFS

题意：给定一些木棒

询问是否能够组成正方形

直接暴力 DFS

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
const int maxn = 24;
int a[ maxn ];
int vis[ maxn ];
int n ,flag;

void dfs( int pos,int len,int LEN,int sum ){//
    if( flag==1 )
        return ;
    if( sum==3 ){
        flag=1;
        return ;
    }
    for( int i=pos;i<=n;i++ ){
        if( vis[ i ]==0 ){
            if( len+a[i]<LEN ){
                vis[ i ]=1;
                dfs( i,len+a[i],LEN,sum );
                vis[ i ]=0;
            }
            else if( len+a[i]==LEN ){
                vis[ i ]=1;
                dfs( 1,0,LEN,sum+1 );
                vis[ i ]=0;
            }
        }
    }
    return ;
}

int main(){
    int t;
    scanf("%d",&t);
    while( t-- ){
        int sum=0;
        scanf("%d",&n);
        for( int i=1;i<=n;i++ ){
            scanf("%d",&a[ i ]);
            sum+=a[ i ];
        }
        if( sum%4!=0 ){
            printf("no\n");
            continue;
        }
        sort( a+1,a+1+n );
        if( a[n]>( sum/4 ) ){
            printf("no\n");
            continue;
        }
        memset( vis,0,sizeof( vis ));
        flag=-1;
        dfs( 1,0,sum/4,0 );
        if( flag!=-1 ){
            printf("yes\n");
        }
        else{
            printf("no\n");
        }
    }
    return 0;
}