[NOIP2008 提高组] 火柴棒等式

题目描述

给你 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 根火柴棍必须全部用上。

输入格式

一个整数 n(1≤n≤24)。

输出格式

一个整数，能拼成的不同等式的数目

#include <bits/stdc++.h>
using namespace std;

const int MAX_NUM = 5000;
const int cost[15] = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6};
int n, cnt[10005], ans;

int main() {
    cin >> n;

    for (int a = 0; a <= 2 * MAX_NUM; a++) {
        if (a >= 10)
            cnt[a] = cnt[a / 10] + cost[a % 10];
        else
            cnt[a] = cost[a];
    }

    for (int a = 0; a <= MAX_NUM; a++)
        for (int b = 0; b <= MAX_NUM; b++)
            if (cnt[a] + cnt[b] + cnt[a + b] + 4 == n)
                ans++;

    cout << ans << endl;

    return 0;
}