题意:
给你一些木棍,看能否用这些木棍围城一个正方形。
思路:
错误思路:
最开始以为只要求出sum的1/4,然后求3遍dfs求木棍和是1/4sum的就行, 但是后来突然发现有一种序列,这样的方法不能使其组成正方形:
1 2 3 4 4 5 6 7
这样如果按照开始的算法,就会产生1 2 5 一组, 然后 4 4 一组,剩下 3 5 6 7 虽然剩下的也是两个1/4sum,但是却不能各自组成1/4sum。
正确思路:
设置一组为一个递归单位,如果一组木棍够一条边了,那就继续找下一组,如果找到了3个边,就返回true,(最后一个不找也知道是不是1/4sum)
但是如果以后的不不能组成两条边,再对本组的进行重新组合。
这样的话,如果开始的组合不能满足解,就可以修改这组的组合,再组成边,然后搜索下一条边,这样就能无遗漏的找出所有的组合,就知道是否能构成正方形了。
刚开始的时候把木棍从小到大排一遍序,这样木棍和一旦和大于边的长度,就可以不用计算后面的木棍。(果断都超了)
Code:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int a[25];
bool v[25];
int n;
bool hasFind;
bool dfs(int u, int h, int l, int sum)
{
if(l == sum)//如果需要找的sum和l一样了,说明找到了3条边了,已经满足。
{
return true;
}
if(h == l) //如果这组的木棍和等于边长了,找下一条边。
{
if(dfs(0, 0, l, sum - l))
{
return true;
}
}
for(int i = u + 1; i < n; i++)
{
if(!v[i])
{
if(h + a[i] <= l)
{
v[i] = true;
if(dfs(i, h + a[i], l, sum))
{
return true;
}
v[i] = false;
}
else //超过边长 直接知道不能组成
{
return false;
}
}
}
return false;
}
int main()
{
int T;
cin >> T;
while(T--)
{
scanf("%d", &n);
int sum = 0, mx = 0;
memset(v, false, sizeof(v));
for(int i = 0; i < n; i++)
{
scanf("%d", &a[i]);
mx = max(mx, a[i]);
sum += a[i];
}
if(sum % 4 != 0 || sum / 4 < mx || n < 4)//如果和不能整出4或者有木棍超出边的长度,就输出no
{
printf("no\n");
continue;
}
int t = sum / 4;
int ans = 0;
sort(a, a + n);
if(dfs(0, 0, t, sum))
{
printf("yes\n");
}
else
{
printf("no\n");
}
}
return 0;
}
/*
1000
8
1 2 3 4 4 5 6 7
*/