算法设计与分析 5.2 洪尼玛的围栏

Posted 2021-03-21 yejifeng

★题目描述

有n块长度不同的木板，他想用这些木板拼成一个等边三角形的围栏。

现在，给你这n块木板的长度，洪尼玛想知道他能否拼成这个围栏？

要求：n块木板都得用上。

★输入格式

第一行为一个正整数T，表示有T组测试数据。

对于每组测试数据，第一行为一个正整数n，表示木板个数；

第二行包含n个正整数，表示每块木板的长度。

★输出格式

如果能拼成围栏输出Yes，否则输出No。

★样例输入

2
4
1 2 3 4
4
1 2 3 3

★样例输出

No
Yes

★提示

对于60%的数据，1<=T<=5，3<=n<=5，每块木板长度大于等于1小于等于100。

对于100%的数据，1<=T<=5，3<=n<=10，每块木板长度大于等于1小于等于100。

★参考代码

#include<bits/stdc++.h>
using namespace std;

int T;
int n;
int side[3],RSide = 0;
int Len[11]; 

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


int dfs(int step){
    if(step==n){
        return side[0]==RSide && side[1]==RSide && side[2]==RSide;
    }
    for(int i=0; i<3; ++i){
        if(side[i]+Len[step]<=RSide){
            side[i]+=Len[step];
            if(dfs(step+1)) return 1;
            side[i]-=Len[step];
        }
    }
    return 0;
}

int fun(){
    scanf("%d",&n);
    int sum=0;
    memset(side, 0, sizeof(side));
    for(int i=0; i<n; ++i){
        scanf("%d", &Len[i]);
        sum += Len[i];
    }
    if(sum%3!=0) return 0;
    RSide = sum/3;
    
    sort(Len, Len+n);
    
    if(Len[0]>RSide) return 0;
    
    return dfs(0); 
}

int main(){
    scanf("%d",&T);
    while(T--){
        fun()==1 ? printf("Yes
") : printf("No
");
    }
    return 0;
}