USACO ORZ（hdu4277 深搜剪枝）

题目链接

题意：  给出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;
}