TYVJ P1012 火柴棒等式 Label：枚舉

背景

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根火柴棍必須全部用上

輸入格式

輸入文件matches.in共一行，又一個整數n（n<=24）。

輸出格式

輸出文件matches.out共一行，表示能拼成的不同等式的數目。

測試樣例1

輸入

【輸入樣例1】?
14?
【輸入樣例2】?
18

輸出

【輸出樣例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>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
int a[10]={6,2,5,5,4,5,6,3,7,6},sum[1005],
    N,M,ans;
int get(int i){
    if(i==0) return 6;
    if(sum[i]>0) return sum[i];
    int pos=0;
    while(i>0){
        pos+=a[i%10];
        i/=10;
    }
    return sum[i]=pos;
}
int main(){
//  freopen("01.txt","r",stdin);
    scanf("%d",&N);
    N-=4;
    for(int i=0;i<=1000;i++){
        if(get(i)>N) continue;
        for(int j=0;j<=1000;j++){
            if(get(i)+get(j)>N) continue;
            if(get(i)+get(j)+get(i+j)==N){
                ++ans;
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}

枚舉到1000，加了點記憶化，（題解說枚舉到800就好）