洛谷 火柴棒等式

[NOIP2008 提高组] 火柴棒等式

题目描述

给你 $n$ 根火柴棍，你可以拼出多少个形如 $A+B=C$ 的等式？等式中的 $A$、$B$、$C$ 是用火柴棍拼出的整数（若该数非零，则最高位不能是 $0$）。用火柴棍拼数字 $0\sim9$ 的拼法如图所示：

注意：

1. 加号与等号各自需要两根火柴棍；
2. 如果 $A\neq B$，则 $A+B=C$ 与 $B+A=C$ 视为不同的等式（$A,B,C\geq0$）；
3. $n$ 根火柴棍必须全部用上。

输入格式

一个整数 $n(1 \leq n\leq 24)$。

输出格式

一个整数，能拼成的不同等式的数目。

样例 #1

样例输入 #1

14

样例输出 #1

2

样例 #2

样例输入 #2

18

样例输出 #2

9

提示

【输入输出样例 1 解释】

$2$ 个等式为 $0+1=1$ 和 $1+0=1$。

【输入输出样例 2 解释】

$9$ 个等式为

$0+4=4$、$0+11=11$、$1+10=11$、$2+2=4$、$2+7=9$、$4+0=4$、$7+2=9$、$10+1=11$、$11+0=11$。

noip2008 提高第二题

思路：
给定一个整数 n，表示可用的火柴棍数量（经过一定调整后的数量），要找出由三个整数组成的组合，使得这三个整数通过某种方式（由 col 函数定义的计算火柴棍数量的方式）所用到的火柴棍总数恰好等于 n，并且满足第一个数与第二个数之和等于第三个数的条件，最后输出满足条件的组合数量。

代码展示：

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 10010;
int n;
int res = 0;
int arr[N];
int nums[10010] = { 6,2,5,5,4,5,6,3,7,6 };//存火柴棍
//计算火柴棍的数量
int col(int x) {
	if (nums[x]>0) return nums[x];
	else  
	{
		int sumFire = 0;
		while (x) {
			sumFire += nums[x % 10];//i%10取个位
			x /= 10;
		}
		return sumFire;
	}
}
//x代表哪个位置，sum记录所用火柴棒总和
void dfs(int x,int sum)
{
	if (sum > n) return;//剪支
	if (x > 3)//当 x > 3 时，意味着已经尝试完了构建一个完整组合所需要的三个位置（即 x 依次取 1、2、3 来分别表示组合中的三个整数位置）。此时就需要对当前构建好的这个可能的组合进行检查，看它是否满足特定的条件。
	{
		if (arr[1] + arr[2] == arr[3]&&sum==n)
		{
			/*for (int i = 1; i <= 3; i++)
			{
				printf("%3d", arr[i]);
			}
			cout << endl;*/
			res++;
		}
		return;
	}

	for (int i = 0; i <= 1000; i++) {
		arr[x] = i;// 将当前位置x的数组元素设置为i，即尝试将i作为组合中当前位置的整数
		dfs(x + 1,sum+col(i));    // 递归调用dfs函数，处理下一个位置（x + 1），并更新所用火柴棍总数为sum + col(i)
		arr[x] = 0;// 将当前位置x的数组元素重置为0，以便在后续尝试其他整数时不受之前设置的影响
    
	}

}

int main()
{
	scanf_s("%d", &n);
	n -= 4;//'+'和‘=’要用掉4根火柴
	dfs(1,0);
	printf("%d\n", res);
	return 0;
}

这个代码在vs上运行没有问题，但在洛谷上却运行失败，想不明白为什么？