搜索|数字三角形

传送门

思路
枚举全排列所有的可能性，然后验证。

在验证的时候，有如下规律：
当n=1时，排列可得到的值为comb[1]
当n=2时，排列可得到的值为comb[1]+comb[2]
当n=3时，排列可得到的值为comb[1]+2 * comb[2]+comb[3]
当n=4时，排列可得到的值为comb[1]+3 * comb[2]+3 * comb[3]+comb[4]
…

实际上，一个排列能够得到的值可以写为：a[1]*comb[1]+…+a[n]*comb[n]
而其中的系数序列a[i]，符合杨辉三角的规律如下：
1 （n=1）
1 1 （n=2）
1 2 1 （n=3）
1 3 3 1 （n=4）
…

因此，在求得杨辉三角矩阵map后，一个排列的值就可以通过上面的公式快速得出，然后进行验证是否等于sum。

此外，在dfs的过程中，需要剪枝，对于必定大于sum的排列，不必枚举出完整的组合，直接返回即可。
Code

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;

int n, sum;
int comb[13];
int vis[13];
int map[13][13];// 保存杨辉三角值

void func(int* arr) {
	int temp = 0;
	for (int i = 1; i <= n - 1; i++) {
		temp += map[n][i] * comb[i];
	}
	int last = sum - temp;
	if (1 <= last && last <= n && vis[last] == -1) {
		for (int i = 1; i <= n - 1; i++) {
			cout << comb[i] << " ";
		}
		cout << last << endl;
		exit(0);
	}
}

void dfs(int depth) {
	if (depth == n) {
		func(comb);
		return;
	}
	// 剪枝
	int temp = 0;
	for (int i = 1; i <= depth - 1; i++) {
		temp += map[n][i] * comb[i];
	}
	if (temp >= sum)
		return;

	for (int i = 1; i <= n; i++) {
		if (vis[i] == -1) {
			vis[i] = 1;
			comb[depth] = i;
			dfs(depth + 1);
			vis[i] = -1;
		}
	}
}

int main() {
	cin >> n >> sum;
	memset(vis, -1, sizeof(vis));
	map[1][1] = 1;
	for (int i = 2; i <= 12; i++) {
		map[i][1] = 1;
		map[i][i] = 1;
		for (int j = 2; j <= i - 1; j++)
			map[i][j] = map[i - 1][j - 1] + map[i - 1][j];
	}
	dfs(1);
	return 0;
}