zufeoj_1128: A+B=C? 洛谷P1149

题目链接:https://www.luogu.org/problemnew/show/P1149

题目描述

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

注意：

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

2. 如果 $A\ne B$ ，则 $A+B=C$ 与 $B+A=C$ 视为不同的等式( $A,B,C>=0$ )

3. $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

n最大为24根。等号和加号需要4根。一个加数至少要2根。还有18根可以选。因为是加数少于和，所以另一个加数至少要8根。

故枚举最大为1111。

#include<bits/stdc++.h>
using namespace std;
int n;
int a[10]={6,2,5,5,4,5,6,3,7,6};
int f(int x){
	if(x==0){
		return a[0];
	}
	//定义此数用到的火柴数量 
	int ans=0;
	//计算数字x每个位置上的火柴棍之和 
	while(x>0){
		ans+=a[x%10];
		x/=10;
	}
	return ans;
}
int main(){
	while(cin>>n){
		int sum=0;
		//可供选择的n=n-(等号和加号所用去的个数) 
		n=n-4;
		for(int i=0;i<=1111;i++){
			for(int j=0;j<=1111;j++){
				//枚举 判断相加能否等于n 
				if(f(i)+f(j)+f(i+j)==n){
					sum++;
				}
			}
		}
		cout<<sum<<endl;
	}
	return 0;
}