能否构成等边三角形（深搜）

Posted 2021-03-12 mld-code-life

#include<iostream>
#include<string.h>
using namespace std;
//编程注意：在全局变量区定义，main函数中初始化，因为变量可能重复使用
int p[100];
bool flag,used[100];
int n,sum;

//三条边，一条边的和s,递归的深度deep
void dfs(int x,int s,int deep){
    //找到一条方案即可结束程序
    if(flag){
        return;
    }
    if(x==3){
        //找到三条边，并且所有木棍都用完
        if(deep==n){
           flag=true;
           cout<<"深度:"<<deep<<endl;
        }
        return;
    }
    //找到一条边时
    if(s==sum/3){
        dfs(x+1,0,deep); //接着原来的深度向下搜。
        return;
    }
    for(int i=0;i<n;i++){
        if(!used[i]){
            used[i]=true; //标记使用，向下递归时不再用
            dfs(x,s+p[i],deep+1);
            used[i]=false; //递归向上返回时取消标记
        }
    }
}
int main()
{
    sum=0,flag=0;
    memset(used,0,sizeof(used));
    for(int i=0;i<n;i++){
        cin>>p[i];
        sum+=p[i];
    }
    //先判断一下是否能被3整数
    if(sum%3!=0){
        cout<<"no"<<endl;
        return 0; //结束程序
    }
    //从0开始搜
    dfs(0,0,0);
    if(flag)
        cout<<"yes"<<endl;
    else
        cout<<"no"<<endl;

    return 0;
}