寒假作业(dfs)

现在小学的数学题目也不是那么好玩的。
看看这个寒假作业：
□ + □ = □
□ - □ = □
□ × □ = □
□ ÷ □ = □
每个方块代表1~13中的某一个数字，但不能重复。
比如：
6 + 7 = 13
9 - 8 = 1
3 * 4 = 12
10 / 2 = 5
以及：
7 + 6 = 13
9 - 8 = 1
3 * 4 = 12
10 / 2 = 5
就算两种解法。（加法，乘法交换律后算不同的方案）
你一共找到了多少种方案？
请填写表示方案数目的整数。
注意：你提交的应该是一个整数，不要填写任何多余的内容或说明性文字。

 思路：直接dfs填充a数组，判断解是否满足条件即可。

代码1.0

#include<bits/stdc++.h>
using namespace std;


int a[15], p = 1;
int v[15];
int cnt = 0;
bool judge() {
	return a[1] + a[2] == a[3] && a[4] - a[5] == a[6] && a[7] * a[8] == a[9] && a[10] == a[11] * a[12];
}
 
void dfs(int s) {
	if (s == 12) {
		if (judge()) {
			cnt++;
			for (int i = 1; i <= 12; i++) {
				cout << a[i] << " ";
			}
			cout << endl;
		}
		return ;
	}
	for (int i = 1; i <= 13; i++) {
		if (!v[i]) {
			a[p++] = i;
			v[i] = 1;
			dfs(s + 1);
			v[i] = 0;
			p--;
		}
	}
} 
 
int main() {
	dfs(0);
	cout << cnt;

	return 0;
}

这个版本的代码效率太低了，在我的电脑上跑了5min才出结果，一开始以为是死循环了，但转念一想dfs不可能出现死循环，最终一定会退出的。于是尝试优化了下代码，果然发现了问题所在。

问题具体出在哪，请接着往下看。

代码2.0

#include<bits/stdc++.h>
using namespace std;

int a[15], p = 1;
int v[15];
int cnt = 0;
 
void dfs(int s) {
	if (s == 4) { // 此时已近填好三个了，直接判断，免得后续再填，过很久才发现不满足要求，效率就低了 
		if (a[1] + a[2] != a[3]) {
			return ;
		}
	}
	if (s == 7) {  
		if (a[4] - a[5] != a[6]) {
			return ;
		}
	}
	if (s == 10) {  
		if (a[7] * a[8] != a[9]) {
			return ;
		}
	}
	if (s == 13) {  
		if (a[10] == a[11] * a[12]) {
			cnt++;
			for (int i = 1; i <= 12; i++) {
				cout << a[i] << " ";
			}
			cout << endl;
		}
	}
	
	
	for (int i = 1; i <= 13; i++) {
		if (!v[i]) {
			a[p++] = i;
			v[i] = 1;
			dfs(s + 1);
			v[i] = 0;
			p--;
		}
	}
} 
 
int main() {
	dfs(1); // dfs(1)填第一层，dfs(2)填第二层，填好三个数后，已经是dfs(4)了 
	cout << cnt;

	return 0;
}

这个版本的代码效率就高多了，运行结果发现，第一个数字由1变成了6，所以将12个数都填好后再判断是否满足条件，这样做的效率是特别低的。

 细心的小伙伴可能发现了两个版本的代码 dfs的起点不一样，一个是dfs(0)，一个是dfs(1)，其实这是无所谓的，就好像数组从0开始存储和从1开始存储是一样个道理。