[数据结构与算法]A - 火柴棒等式

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

用火柴棍拼数字 0-9，0?9 分别用的火柴棒数量为：6 2 5 5 4 5 6 3 7 6；
注意：

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

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

n 根火柴棍必须全部用上

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

输出格式
输出能拼成的不同等式的数目。

Sample Input
5
Sample Output
0

#include <iostream>
#include<algorithm>
using namespace std;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int a[10] = { 6,2,5,5,4,5,6,3,7,6 };//0~9需要的火柴棒数
    int sum = 0;
    int n;
    cin >> n;
    n -= 4;
    if (n < 9) {//最少的火柴棒组合是1+1=2，去掉运算符需要9根火柴
        cout << 0;
        return 0;
    }
    int ti, tj, tk, t;//分别记录数字i，j，他们的加和k大于10的部分，总计 的火柴棒数
    for (int i = 0;; i++) {
        for (int j = 0;; j++) {
            ti = tj = tk = t = 0;//归0
            int k = i + j;
            int lowi = i, lowj = j, lowk = k;//拷贝，方便下面计算
            while (lowi >= 10) {
                int u = lowi % 10;
                ti += a[u];
                lowi /= 10;
            }
            while (lowj >= 10) {
                int u = lowj % 10;
                tj += a[u];
                lowj /= 10;
            }
            while (lowk >= 10) {
                int u = lowk % 10;
                tk += a[u];
                lowk /= 10;
            }
            if (ti + a[lowi] >= 6 + n) {//当数字i需要的火柴棒数大于n时，结束程序
                cout << sum;            //之所以用2*n，是因为i大不代表用的火柴棒多，多1个足够(洛谷上足够，但vj上不够，   
                return 0;               //我试了试,时间超限的临界是n=30，再多就会超限，这就多加6了)，j同理
            }
            if (tj + a[lowj] >= 6 + n) break;//当数字j需要的火柴棒数大于n时，进入下一个i，同时归0j
            t += ti + tj + tk;//不用ti，tj，tk代表数总共需要的火柴棒数是为了避免0的问题
            t += a[lowk] + a[lowi] + a[lowj];
            if (t == n) {
                sum++;//总火柴棒数等于n的时候，累和加一
            }
            else continue;//否则进入下一个j
        }
    }
}