题目链接:
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;
}