洛谷P1136火柴棒等式

题目描述：
给你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根火柴棍必须全部用上；

输入样例：
18
输出样例：
9

#include<stdio.h>
int a[2001]={6};
int nu[10] = {6,2,5,5,4,5,6,3,7,6};
int main()
{
    int m, n;
    int total = 0;
    scanf("%d", &n);
    for(int i = 1; i <= 2000; i++)
    {
        m = i;
        while(m > 0)//求每个数所用的火柴棒
        {
        	a[i] += nu[m % 10];
            m /= 10;
        }
    }
    for(int i = 0; i <= 1000; i++)
    {
        for(int j = 0; j <= 1000; j++)
        {
        	if(a[i] + a[j] + a[i+j] + 4 == n)//加号与等号需要4根火柴 
			{
				total++;//方案数加1 
			}
		}
    }
    printf("%d", total);
    return 0;
}