洛谷 P1149 [NOIP2008 提高组] 火柴棒等式 dfs C语言

题目链接：

https://www.luogu.com.cn/problem/P1149

该题有两个问题需要解决。

第一个问题，选出的3个数要满足等式，我写的是靠搜索出来的三个数字存入arr数组，例如  用arr[1] + arr[2] = arr[3]。火柴可以组成最大数组并且满足题目条件，我假设最大为800.

第二个问题，这三个数字的火柴数量以及 + 和 = （四个火柴） 之和要等于输入的n。将n = n - 4     现在的n就等于三个数字的火柴数量。数字火柴的数量可以拆开来计算之和，用函数计算后返回。

#include<iostream>
#include<algorithm>
using namespace std;
int n;
int num[1000] = {6,2,5,5,4,5,6,3,7,6};//储存初始化0-9数字的火柴数量 
int arr[10]; //储存临时的三个数字 
int res = 0;//记录方案 

int cal(int k)//计算火柴数量
{
	if(num[k] > 0)
	return num[k];
	else
	{
		int sumFire = 0;
		while(k)
		{
			sumFire = sumFire + num[k%10];//将组成为k数字的火柴数量拆开计算相加 
			k = k / 10;
		}
		return sumFire;
	}
	
}

void dfs(int x,int sum)
{
	if(sum > n)//剪枝
	return;

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

	for(int i = 0 ; i <= 800 ; i++)//假设火柴棍满足题目条件可以组成的数字在800之前 
	{
		arr[x] = i;
		dfs(x+1,sum + cal(i));
		arr[x] = 0; 
	}
}

int main(void)
{
	cin >> n;
	n = n-4;
	dfs(1,0);
	cout << res << endl;
	return 0;
}