P1149 火柴棒等式

题目链接：
P1149 火柴棒等式

题目描述

给你n根火柴棍，你可以拼出多少个形如“A+B=C”的等式？等式中的A、B、C是用火柴棍拼出的整数（若该数非零，则最高位不能是0）。用火柴棍拼数字0−9的拼法如图所示：

注意：

• 加号与等号各自需要两根火柴棍
• 如果A≠B，则A+B=C与B+A=C视为不同的等式(A,B,C>=0)
• n根火柴棍必须全部用上

输入格式

一个整数n(n <= 24)

输出格式

一个整数，能拼成的不同等式的数目。

输入输出样例

输入 #1

14

输出 #1

2

输入 #2

18

输出 #2

9

说明/提示

样例 #1

2 个等式为：
0 + 1 = 1
1 + 0 = 1

样例 # 2

9 个等式为：
0+4=4
0+11=11
1+10=11
2+2=4
2+7=9
4+0=4
7+2=9
10+1=11
11+0=11

解析

一开始读完这道题，想到的是先拼数字，再组成公式。后来发现先组公式，在看能不能拼出来简单。
等式中的加号和等于号也要用火柴棒，所以一开始要先减去4个。之后，先把1-999都需要几个火柴棒拼放到一个数组中，然后用递归，每一层寻找 x1 + x2 = x3 中的一个变量，找到第三层时，判断火柴棒是否够用，并且是否能拼出等式，如果能，结果加1，如果不能，就回溯，回溯的时候记得消除影响。

代码

#include<stdio.h>
int nn[10] = {6,2,5,5,4,5,6,3,7,6};//0-9所需要的火柴棒
int l[4];
int num[999] = {6};//0-998所需要的火柴棒
int n;
int ans;

void dfs(int def){
    for(int i = 0;i <= 999;i++){
        if(n - num[i] >= 0){//火柴棒如果还够用
            l[def] = i;
            n -= num[i];
            if(def == 3){
                //是否能拼出等式并且火柴棒够用
                if(l[1] + l[2] == l[3]&&n == 0) ans++;
            }else{
                dfs(def + 1);
            }
            n += num[i];
        }
    }
}

int main(){
    scanf("%d",&n);
    n -= 4;
    //求出0-998需要的火柴棒
    for(int i = 0;i <= 999;i++){
        int j = i;
        while(j >= 1){
            num[i] = num[i] + nn[j % 10];
            j /= 10;
        }
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}