HDU 1518 Square

RedPolya

于 2016-12-02 11:07:54 发布

阅读量307

点赞数

CC 4.0 BY-SA版权

分类专栏： HDU 文章标签： dfs hdu 1518 c++

本文链接：https://blog.youkuaiyun.com/RedPolya/article/details/53432717

HDU 专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种通过回溯搜索算法来判断一组数字是否能够组成正方形四边的解决方案。该算法首先对输入的数字进行排序并检查总和是否为4的倍数，然后使用递归函数尝试组合数字以形成正方形的四条相等边。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Square

给一串数字，判断是否能凑成正方形的四个边。

裸搜题，不过没有一遍AC。。。。

直接上代码吧。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>

using namespace std;

int T,n,sum,len;
int info[25];
bool v[25],flag;

int dfs(int id,int completed,int length){
    if(completed==4)return 1;
    for(int i=id;i<=n;i++){
        if(!v[i]){
            if(length-info[i]==0){
                v[i]=1;
                if(dfs(1,completed+1,len)) return 1; 
                v[i]=0;
            }
            else if(length-info[i]>0){
                v[i]=1;
                if(dfs(i+1,completed,length-info[i])) return 1;
                v[i]=0;
            }
        }
    }
    return 0;
}

bool cmp(int a,int b){
    return a>b;
}

int main(){
    cin>>T;
    while(T--){
        sum = 0;
        scanf("%d",&n); 
        for(int i=1;i<=n;i++){
            scanf("%d",info + i);
            sum+=info[i];
        }
        sort(info+1,info+1+n,cmp);
        if(sum%4||n<4){
            puts("no");
            continue;
        }
        len = sum/4;
        memset(v,0,sizeof(v));
        if(dfs(0,0,len))    puts("yes");
            else    puts("no");
    }
    return 0;
}