洛谷 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

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

这道题看起来好难啊，但实际上就是一道模拟题~

蠢蠢的我刚开始竟然想dfs……后来才知道可以这样写……学习了……

一定要考虑为0时的情况！

#include<cstdio>

int a[10]={6,2,5,5,4,5,6,3,7,6},n,tot;

int num(int u)
{
	int ans=0;
	if(!u) return a[0];
	while(u)
	{
		ans+=a[u%10];u/=10;
	}
	return ans;
}

int main()
{
	scanf("%d",&n);n-=4;
	for(int i=0;i<=1111;i++)
	  for(int j=0;j<=1111;j++)
	    if(num(i+j)+num(i)+num(j)==n) tot++;
	printf("%d\n",tot);
	return 0;
}