洛谷 P1149 [NOIP2008 提高组] 火柴棒等式（指数型枚举+预处理）

题目链接:P1149 [NOIP2008 提高组] 火柴棒等式 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 10010;
int n,res=0;  //res记录方案数
int arr[N]; //每一位数
int sum[N] = { 6,2,5,5,4,5,6,3,7,6 }; //每个数所需的火柴棍数

void dfs(int x,int ans)
{
	if (ans > n) return; //减枝，不然TLE

	if (x > 3)
	{
		if (arr[1] + arr[2] == arr[3] && ans == n)
			res++;

		return;
	}
	
	for (int i = 0; i <= 1000; i++)
	{
		arr[x] = i;
		dfs(x + 1, ans + sum[i]);
		arr[x] = 0;
	}
}
int main()
{
	cin >> n;
	n -= 4;
	for (int i = 10; i <= 1000; i++)
		sum[i] = sum[i / 10] + sum[i % 10]; //预处理火柴棍数
	dfs(1,0);
	cout << res << endl;
	return 0;
}