[洛谷]P1149 火柴棒等式 (#枚举搜索 -1.7)

题目描述

给你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：
14
样例输入2：
18

样例输出1：
2
样例输出2：
9

---

思路

枚举a和b，c可以通过a和b算出来，然后写一个外部函数。

#include <stdio.h>
#include <iostream>
using namespace std;
int stick[11]={6,2,5,5,4,5,6,3,7,6};//0-9需要用多少根火柴棒 
int search(int x)
{
	int num(0);
	while(x/10!=0)
	{
		num+=stick[x%10];//得到x的末尾数字，并将这个数所需要的火柴棒累计 
		x=x/10;//去末尾数字 
	}
	num+=stick[x];//最后再加，此时x是个位数 
	return num;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int i,j,k,n,s(0);
	cin>>n;
	for(i=0;i<1111;i++)//枚举a和b 
	{//为什么到1111？因为A+B=C这个式子中任何一个数都不会超过1111。 
		for(j=0;j<1111;j++)
		{
			k=i+j;//算C 
			if(search(i)+search(j)+search(k)==n-4)//除去+和=，还剩下n-4根火柴棒 
			{
				//cout<<i<<"+"<<j<<"="<<k<<endl;
				s++;
			}
		}
	}
	cout<<s<<endl;
	return 0;
}