洛谷P1149 [NOIP2008 提高组] 火柴棒等式

#include<iostream>
using namespace std;
int num[] = {6,2,5,5,4,5,6,3,7,6};//每个数需要的火柴棒数
int n;
int ans,ans1;
void sear(int x){
	int i,j;
	for(i = 0; i <= 1000; ++i){//最大是3位数，不到1000的话只能拿60分
		for(j = 0;j <= 1000 - i; ++j){
		int num1 = 0,num2 = 0,num3 = 0;
			int a = i;
			int b = j;
			int c = i + j;
			do{//记录数用的火柴棒
				num1 += num[a%10];
				a /= 10;
			}while(a != 0);
			do{
				num2 += num[b%10];
				b /= 10;
			}while(b != 0);
			do{
				num3 += num[c%10];
				c /= 10;
			}while(c != 0);
			if(num1 + num2 + num3 == x)ans++;
			if(i == j && num1 + num2 + num3 == x)ans1++;//如果有相同的A,B就加1
		}
	}
}
int main(){
	cin >> n;
	sear(n-4);
	cout << ans-ans1/2;//所得到的总数-重复记的数
	return 0;
}