火柴棒等式

总时间限制: 1000ms 内存限制: 65536kB
描述
给你n根火柴棍，你可以拼出多少个形如“A+B=C”的等式？等式中的A、B、C是用火柴棍拼出的整数（若该数非零，则最高位不能是0）
注意：

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

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

3. n根火柴棍必须全部用上

输入
输入一个整数n（n≤24）。
输出
输出能拼成的不同等式的数目。
样例输入
5
样例输出
0
这道题我是先把从0到2000每个数字所用的火柴棒数都算出来，然后一个循环搞定。刚开始好像想复杂了……
代码如下：

#include<bits/stdc++.h>
using namespace std;
int p[2001]={6};
int main()
{
	int n,ans=0;
	cin>>n;
	int shu[10]={6,2,5,5,4,5,6,3,7,6};//从0到9所用的火柴棒数量。
	for(int i=0;i<2000;i++)
	{
		int j=i;
		while(j>=1)
		{
			p[i]=p[i]+shu[j%10];
			j=j/10;
		}
	}
	for(int i=0;i<1000;i++)
	for(int j=0;j<1000;j++)
	{
		if(p[i]+p[j]+p[i+j]+4==n)
		{
			ans++;
		}
	}
	cout<<ans<<endl;
	return 0;
	}