拼多多笔试 2020/5/6

描述
给定一组不同长度的棒, 是否有可能加入它们端到端形成一个正方形？

输入
第一行输入包含 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;
		}

	}
}