【搜索】洛谷 P1149 火柴棒等式

题目描述

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

注意：

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

如果A≠B，则A+B=C与B+A=C视为不同的等式（A、B、C>=0）

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

代码

#include<iostream>
using namespace std;
int g[10]={6,2,5,5,4,5,6,3,7,6};
int main()
{
    int s=0,n,a=0,b=0,c=0;
    cin>>n;
    for(int i=0;i<=1000;i++)
    {
        for(int j=0;j<1000;j++)
        {
            int k=i,p=j,m=i+j;
            if(k==0)a=6;
            else while(k>0){a+=g[k%10];k=k/10;}
            if(p==0)b=6;
            else while(p>0){b+=g[p%10];p=p/10;}
            if(m==0)c=6;
            else while(m>0){c+=g[m%10];m=m/10;}
            if((a+b+c)==(n-4))s++;
            a=0;b=0;c=0;
        }
    }
    cout<<s;
    return 0;
}