计蒜客 等边三角形 dfs

题目：

https://www.jisuanke.com/course/2291/182238

思路：

1.dfs(int a,int b,int c,int index)//a,b,c三条边的边长,index是数组的下标

从dfs(0,0,0,0）开始，每次选的长度可能加到a,也可能b，也可能c，

三种可能分别dfs:

dfs(a+len[index+1],b,c,index+1);
dfs(a,b+len[index+1],c,index+1);
dfs(a,b,c+len[index+1],index+1);

2.剪枝

数据太大的边直接结束递归，此题中因为是求等边三角形，对于所有边的和sum，除以三的值肯定是大于等于abc三条边的。

3.提取重复的运算

一开始我sum没有/3，而是在dfs里面/3判断，所以就TLE了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
using namespace std;
//计蒜客 等边三角形
int len[25],n;
bool ok;
int sum;
void dfs(int a,int b,int c,int index)//a,b,c三条边的边长,index是数组的下标 
{
    if(a>sum||b>sum||c>sum)//剪枝 
    return;
    if(index>n)
        return;
    if(a==b&&b==c&&index==n)
    {
        ok=true;
        return;
    }
    
    if(ok)
    return;
    
    {
        dfs(a+len[index+1],b,c,index+1);
        dfs(a,b+len[index+1],c,index+1);
        dfs(a,b,c+len[index+1],index+1);
    }
}
int main()
{
    while(scanf("%d",&n)==1)
    {
        sum=0;
        for(int i=1;i<=n;++i)
        {
            scanf("%d",&len[i]);
            sum+=len[i];
        }
        ok=false;
        sum/=3;
        dfs(0,0,0,0);
        if(ok)
        printf("yes\n");
        else
        printf("no\n");
    }
    return 0;    
}