洛谷 P1149 火柴棒等式

https://www.luogu.com.cn/problem/P1149
题目描述
给你n根火柴棍，你可以拼出多少个形如“A+B=CA+B=C”的等式？等式中的AA、BB、CC是用火柴棍拼出的整数（若该数非零，则最高位不能是00）。用火柴棍拼数字0-90−9的拼法如图所示：

注意：

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

如果A≠B，则A+B=C与B+A=C视为不同的等式(A,B,C>=0)

n根火柴棍必须全部用上

输入格式
一个整数n(n<=24)。

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

输入输出样例
输入
14
输出
2
输入
18
输出
9

说明/提示
【输入输出样例1解释】
2个等式为0+1=10+1=1和1+0=11+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>
#include<cstdio>

using namespace std;

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

int fun(int x)
{
    int num = 0;
    while (x/10!=0)//不能用while（x），要不然0不能进行计算
    {
        num += a[x % 10];
        x /= 10;
    }
    num += a[x];
    return num;
}

int main()
{
    int t, i, j, k, sum;
    while (scanf("%d", &t) != EOF)
    {
        sum = 0;
        for (i = 0; i <= 1111; i++)//等式中任意一个数都不能超过1111
        {
            for (j = 0; j < 1111; j++)
            {
                k = i + j;
                if (fun(i) + fun(j) + fun(k) + 4 == t) 
                {
                    //printf("%d+%d=%d\n", i, j, k);
                    sum++;
                }
            }
        }
        printf("%d\n", sum);
    }
    return 0;
}