题意: 给出n个数, 用这些数组成三个数, 问能够拼成三角形的数目
思路: 深搜, 将这n个数分成三堆再判断是否组成三角形, 但是肯定会爆掉, 所以中间加上set判重剪枝。 虽然在中间进行判重 代码比较繁琐, 但是运行效率还是很可观的。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <queue>
#include <set>
using namespace std;
int num[20];
int n;
int cnt;
int sum;
int max(int l1, int l2, int l3)
{
return l1>l2? (l1>l3?l1:l3) : (l2>l3?l2:l3);
}
int min(int l1, int l2, int l3)
{
return l1<l2? (l1<l3?l1:l3) : (l2<l3?l2:l3);
}
struct node
{
int x, y, z;
};
struct compare
{
bool operator()(node a, node b)
{
if(a.x > b.x)
return true;
if(a.x < b.x)
return false;
if(a.y > b.y)
return true;
if(a.y < b.y)
return false;
if(a.z > b.z)
return true;
return false;
}
};
set <node, compare> set1;
void dfs(int l1, int l2, int l3, int j)
{
int mx, mi,s1;
node p;
if(j == n)
{
int maxn = max(l1, l2, l3);
if(l1+l2+l3-maxn > maxn && l1 > 0 && l2 > 0 && l3 > 0)
cnt ++;
}
else
{
if(l1 + num[j] < (sum+1)/2)
{
mx = max(l1+num[j], l2, l3);
mi = min(l1+num[j], l2, l3);
p = {mx,l1+num[j]+l2+l3-mi-mx,mi};
s1 = set1.size();
set1.insert(p);
if(set1.size() > s1)
dfs(l1+num[j], l2, l3, j+1);
}
if(l2 + num[j] < (sum+1)/2)
{
mx = max(l1, l2+num[j], l3);
mi = min(l1, l2+num[j], l3);
p = {mx,l1+num[j]+l2+l3-mi-mx,mi};
s1 = set1.size();
set1.insert(p);
if(set1.size() > s1)
dfs(l1, l2+num[j], l3, j+1);
}
if(l3 + num[j] < (sum+1)/2)
{
mx = max(l1, l2, l3+num[j]);
mi = min(l1, l2, l3+num[j]);
p = {mx,l1+num[j]+l2+l3-mi-mx,mi};
s1 = set1.size();
set1.insert(p);
if(set1.size() > s1)
dfs(l1, l2, l3+num[j], j+1);
}
}
}
int main()
{
int t;
scanf("%d", &t);
while(t--)
{
int i;
scanf("%d", &n);
sum = 0;
for(i = 0;i < n; i++)
{
scanf("%d", &num[i]);
sum += num[i];
}
cnt = 0;
set1.clear();
dfs(0, 0, 0, 0);
cout << cnt << endl;
}
return 0;
}