码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2019-12-22 16:41:59      阅读:95      评论:0      收藏:0      [点我收藏+]

标签:设计   fun   mes   memset   cpp   code   printf   算法   输入格式   

★题目描述

有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\n") : printf("No\n");
    }
    return 0;
}

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

标签:设计   fun   mes   memset   cpp   code   printf   算法   输入格式   

原文地址:https://www.cnblogs.com/yejifeng/p/12080005.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!