标签:设计 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;
}
标签:设计 fun mes memset cpp code printf 算法 输入格式
原文地址:https://www.cnblogs.com/yejifeng/p/12080005.html