蒜头君手上有一些小木棍,它们长短不一,蒜头君想用这些木棍拼出一个正方形,并且每根木棍都要用到。 例如,蒜头君手上有长度为 11,22,33,33, 33 的 55 根木棍,他可以让长度为11,22 的木棍组成一条边,另外三根分别组成 33 条边,拼成一个边长为 33 的正方形。蒜头君希望你提前告诉他能不能拼出来,免得白费功夫。
输入格式
首先输入一个整数 n(4 \le n \le 20)n(4≤n≤20),表示木棍数量,接下来输入 nn 根木棍的长度 p_i(1 \le p_i \le 10000)pi(1≤pi≤10000)。
输出格式
如果蒜头君能拼出正方形,输出"Yes",否则输出"No"。
样例输入1
4 1 1 1 1
样例输出1
Yes
样例输入2
5 10 20 30 40 50
样例输出2
No
ac
#include <bits/stdc++.h>
using namespace std;
int n,l[25],sum=0,t=0;
bool cmp(int a,int b)
{
return a>b;
}
void dfs(int k,int l1,int l2,int l3,int l4)//代表将小标为k的选择性放入l1,l2,l3,l4;
{
if(l1>sum||l2>sum||l3>sum||l4>sum)
return ;
if(l1==sum&&l2==sum&&l3==sum&&l4==sum)
{
cout<<"Yes";
t=1;
exit(0);
return ;
}
if(k==n)//到达深搜终点
return ;
dfs(k+1,l1+l[k],l2,l3,l4);
dfs(k+1,l1,l2+l[k],l3,l4);
dfs(k+1,l1,l2,l3+l[k],l4);
dfs(k+1,l1,l2,l3,l4+l[k]);
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
{
cin>>l[i];
sum+=l[i];
}
if(sum%4)//代表不能整除 直接排除
{
cout<<"No";
return 0;
}
sum/=4;
sort(l,l+n,cmp);
dfs(0,0,0,0,0);
if(!t)
cout<<"No";
return 0;
}