洛谷P1149 火柴棒等式

题目描述

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

注意：

1. 加号与等号各自需要两根火柴棍
2. 如果$A\ne B$，则$C$与$=C$视为不同的等式($A,B,C>=0$)
3. n根火柴棍必须全部用上

输入输出格式

输入格式：

一个整数n(n<=24)。

输出格式：

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

输入输出样例

输入样例#1：

14

输出样例#1：

2

输入样例#2：

18

输出样例#2：

9

说明

【输入输出样例1解释】

$2$个等式为$=1$和$=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

TIPS：

从正面想n根火柴棒能拼出几个成立的等式比较麻烦，不如将成立的等式i + j = k所需的火柴棒根叔储存在数组sav[i][j]中，之后搜索就行了。如何确定循环上界是个问题。由于需要最少的火柴棒拼出的数字为1，需要2根，而24 / 2 / 3 = 4，所以循环到1111应该就能满足n<=24的要求。

• Answer

#include<cstdio>
#include<iostream>
#define SIZE 1111
using namespace std;
int dig[10] = {6,2,5,5,4,5,6,3,7,6};
int sav[SIZE][SIZE];
int num(int n);
int main(void){
	int n, ans = 0;	
	for(int i = 0; i < SIZE; i++){
		for(int j = 0; j < SIZE; j++){
			int k = i + j;
			sav[i][j] = num(i) + num(j) + num(k) + 4;//组成 i + j = k需要多少根火柴棒 
		}
	}
	while(cin >> n){
	ans = 0;
	for(int i = 0; i < SIZE; i++){
		for(int j = 0; j < SIZE; j++){
			if(sav[i][j] == n)
			ans++; 
		}
	}
	cout << ans << endl;
	}
	return 0;
}
int num(int n){
	if (n == 0)
		return dig[0];
	int sum = 0;
	while(n){
		sum = sum + dig[n % 10];
		n /= 10;
	}
	return sum;
}