描述
给定一组不同长度的棒, 是否有可能加入它们端到端形成一个正方形?
输入
第一行输入包含 N, 测试用例的数量。每个测试用例都以一个整数 4 < = M < = 20 开头, 即棍子的数量。M 整数跟随;每个都给出一根棍子的长度--介于1和1万之间的整数。
输出
对于每种情况, 输出一条包含 "是" 的行, 如果是可能形成一个正方形;否则输出 "否"。
示例输入
3
4 1 1 1 1
5 10 20 30 40 50
8 1 7 2 6 4 4 3 5
示例输出
yes
no
yes
#include<iostream>
#include<vector>
#include<algorithm>
#include<string>
#include <string.h>
using namespace std;
int a[10001];
vector<int>l;
int sum;
int n;
int t;
bool search(int side, int side_left, int mark)
{
cout <<"sum="<< sum << endl;
if (side == 4)
{
return true;
}
else
{
for (int i = mark; i < n; i++)
{
if (a[i] == 0)
{
a[i] = 1;
if (side_left == l[i])
{
if (search(side + 1, sum / 4, 0))
{
return true;
}
}
else if (l[i] < side_left)
{
if (search(side, side_left - l[i], i + 1))
{
return true;
}
}
a[i] = 0;
}
}
}
return false;
}
int main()
{
cin >>t;
for (int i = 0; i < t; i++)
{
cin >> n;
int x;
sum = 0;
int maxl = 0;
for (int j = 0; j < n; j++)
{
cin >> x;
l.push_back(x);
sum += x;
maxl = max(x, maxl);
}
if (sum % 4 != 0 || maxl > sum / 4)
{
cout << "NO" << endl;
continue;
}
memset(a, 0, sizeof(a));
if (search(1, sum / 4, 0))
{
cout << "YES" << endl;
}
else
{
cout << "NO" << endl;
}
}
}