递归/DFS学习6

[NOIP2008 提高组] 火柴棒等式

题目描述

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

注意：

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

输入格式

一个整数$n(1\le n\le 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$。

思路：

本题是排列枚举，可以出现0+0=0的情况（如果允许的情况），我们‘可以选选出3个数，并且将火柴数加入搜索中，看看是否符合火柴数-4（‘+’和‘=’要分别要两个火柴）且前两个数之和是否等于第三个，由于数字是由火柴数组成的，还需要写一个函数来计算选出数的火柴数，由于数据量比较大，还需要剪枝，如果火柴数已经大于要求，直接return

#include <bits/stdc++.h>
using namespace std;
int n,ans=0;
vector<int> re(1e3);
vector<int> v(3);
int val(int n){//可优化
	int val=0;
	if(n>9){
		int t=n;
		while(t){
			int t1=t%10;
			t/=10;
			val+=re[t1];
		}
	}else{
		val=re[n];
	}
	return val;
}
void dfs(int x,int cnt){
	if(cnt>n-4){
		return;
	}
	if(x>2){
		if(v[0]+v[1]==v[2]&&cnt==n-4){
			ans++;
		}
		return;
	}
	for(int i=0;i<=1000;i++){
		v[x]=i;
		dfs(x+1,cnt+val(i));
	}
}
int main() {
	cin >> n;
	re={6,2,5,5,4,5,6,3,7,6};
	dfs(0,0);
	cout << ans;
	return 0;
}

优化：

其实val的计算可以采用更优的方法，因为val的值计算有可能会重复，我们可以计算剩下的re元素的值，这样就不会重复计算数字所需的火柴数量了，我们可以通过递推来计算如11,12等火柴数

代码如下：

for (int i = 10; i <= 1000; i++) {//递推
		re[i] = re[i / 10] + re[i % 10];
	}

所以我们最优代码为：

#include <bits/stdc++.h>
using namespace std;
int n, ans = 0;
vector<int> re(1e3);
vector<int> v(3);
void dfs(int x, int cnt) {
	if (cnt > n - 4) {
		return;
	}
	if (x > 2) {
		if (v[0] + v[1] == v[2] && cnt == n - 4) {
			ans++;
		}
		return;
	}
	for (int i = 0; i <= 1000; i++) {
		v[x] = i;
		dfs(x + 1, cnt + re[i]);
	}
}
int main() {
	cin >> n;
	re = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6};
	for (int i = 10; i <= 1000; i++) {//递推
		re[i] = re[i / 10] + re[i % 10];
	}
	dfs(0, 0);
	cout << ans;
	return 0;
}