luogu P1149 火柴棒等式

题目描述

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

注意：

1. 加号与等号各自需要两根火柴棍

2. 如果A≠B，则A+B=C与B+A=C视为不同的等式（A、B、C>=0）

3. n根火柴棍必须全部用上

输入输出格式

输入格式：

 

输入文件matches.in共一行，又一个整数n（n<=24）。

 

输出格式：

 

输出文件matches.out共一行，表示能拼成的不同等式的数目。

 

输入输出样例

输入样例#1：

样例输入1：
14
样例输入2：
18

输出样例#1：

样例输出1：
2
样例输出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

顺便敲了发搜索
因为给的数据范围：24根火柴棒因为加号和等于号所以只有20根有用，最多两个到1000
然后爆搜

#include<cstdio>
using namespace std;
int num[2333]={6,2,5,5,4,5,6,3,7,6};
int ans=0;int n;
int tmp[2333];
void dfs(int x) {
    for(int i=0;i<=999;++i) {
        if(n-num[i]>=0){
            tmp[x]=i;
            n-=num[i];
            if(x==3) {
                if(tmp[1]+tmp[2]==tmp[3]&&n==0)ans++;
            }
            else dfs(x+1);
            n+=num[i];
        }
    }
}
int main () {
    
    scanf("%d",&n);n-=4;
    for(int i=10;i<=999;++i) {
        num[i]=num[i/10]+num[i%10];//递推 
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}