HDU5937Equation(暴搜+剪枝)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5937
题意：

给出1-9的个数，然后组成a+b=c的式子(1<=a,b,c<=9),这样的式子有多少个？

思路：

算一下，有36个，满36个需要用到的a[i]的个数是17-i,先判满不满，a[i]=min(a[i],17-i)，多于限制个没什么用，然后只要暴搜+剪枝

#include<bits/stdc++.h>
using namespace std;
int a[10], d[40][3], vis[12], ans;
void init()
{
    int cnt = 0;
    for(int i = 1; i <= 9; i++)
        for(int j = 1; j +i <= 9; j++)
        {
            d[cnt][0] = i;
            d[cnt][1] = j;
            d[cnt++][2] = i+j;
            vis[i]++;vis[j]++;vis[i+j]++;
        }
}
bool check(int x)
{
    if(a[d[x][0]] < 0 || a[d[x][1]] < 0 || a[d[x][2]] < 0) return false;
    return true;
}
void add(int x, int val)
{
    a[d[x][0]] += val;
    a[d[x][1]] += val;
    a[d[x][2]] += val;
}
void dfs(int deep, int cnt)
{
    if(36 - deep + cnt <= ans || deep == 36) return ;
    add(deep, -1);
    if(check(deep))
    {
        ans = max(ans, cnt+1);
        dfs(deep+1, cnt+1);
    }
    add(deep, 1);
    dfs(deep+1, cnt);
}
int main()
{
    int t;
    init();
    scanf("%d", &t);
    for(int kase = 1; kase <= t; kase++)
    {
        bool ok = true;
        for(int i = 1; i <= 9; i++)
        {
            scanf("%d", &a[i]);
            a[i] = min(a[i], 17 - i);
            if(a[i] < 17 - i) ok = false;
        }
        if(ok)
            printf("Case #%d: %d\n", kase, 36);
        else
        {
            ans = 0;
            dfs(0, 0);
            printf("Case #%d: %d\n", kase, ans);
        }
    }
    return 0;
}