火柴棒等式c++

先建立一个sum数组，打表存入1—9每个数字需要的火柴棒数，然后手动二重循环0—1000的所有数字，写一个int型函数用来计算每个数字需要多少根火柴棒（当前数字%10后在sum数组的下标），然后，最后返回

如果数字A+数字B+数字C+4（等号和加号）==总火柴棒数，计数器++;

#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <ctime>
using namespace std;
int sum[11] = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6};
int n,t = 0;
int work(int x) {
    int ans = 0, k;
    if(x == 0) return sum[0];
    else {
        while(x != 0) {
            k = x % 10;
            x /= 10;
            ans += sum[k];
        }
        return ans;
    }
}
int main() {
    cin >> n;
    for(int i = 0; i <= 1000; i++)
        for(int j = 0; j <= 1000; j++) {
            if(work(i) + work(j) + work(i + j) + 4 == n)
                t++;
        }
    cout << t << '
';
    return 0;
}