[USACO06FEB] Backward Digit Sums G/S——洛谷pP1118

目录

 [USACO06FEB] Backward Digit Sums G/S

 题面翻译

输入格式

输出格式

问题分析

---

 [USACO06FEB] Backward Digit Sums G/S

 题面翻译

有这么一个游戏：

• 写出一个 1~ n的排列 ai，然后每次将相邻两个数相加，构成新的序列，再对新序列进行这样的操作，显然每次构成的序列都比上一次的序列长度少 1，直到只剩下一个数字位置。

下面是一个例子：

- 3,1,2,4；
- 4,3,6；
- 7,9；
- 16。

• 最后得到 16 这样一个数字。
• 现在想要倒着玩这样一个游戏，如果知道 n，以及最后得到的数字的大小 sum，请你求出最初序列 a_i（应该是一个 1~ n 的排列）。若答案有多种可能，则输出字典序最小的那一个。

输入格式

共一行两个正整数 n,sum。

输出格式

输出包括一行，为字典序最小的那个答案。

当无解的时候，请什么也不输出。

样例 

• 样例输入 

```
4 16
```

• 样例输出 

```
3 1 2 4
```

问题分析

本题用暴力法会超时，对1~n这n个数全排列完了之后还要进行三角运算判断是否等于sum，对每个序列三角运算运算次数为1+2+3+4+5+.....+n=,时间复杂度为O(),那么总复杂度就为O(n!),显然会超时

• 如果可以找到第一行与最后一行的关系那么问题就很简单了，不妨设原序列为a,b,c,d,然后算下去就可以发现sum=1*a+3*b+3*c+1*d，观察系数会发现这就是杨辉三角，那么我们只需要事先计算好1~n层的杨辉三角，就可以轻松得到系数,到此算法优化为O(n!n)
• 之后就是剪枝了，有句话说的好没有剪枝的dfs不是好的dfs算法(bushi),确实，如果dfs没有剪枝和暴力法其实没有区别，本题使用的时最优化剪枝

#include<iostream>
#include <cstring> 
//要使用memset需要导入cstring，我的vs自带了，搞得wrong好几次（草率了）
using namespace std;

int n, sum,flag=0;   //flag用于控制程序结束
bool vis[20];  //记录该数字是否被用过，这里所求的序列是不允许重复的
int c[20];     //存储答案
int a[13][13];  //存储杨辉三角

//初始化杨辉三角
void init() {
	memset(a, 0, sizeof(a));
	memset(vis, false, sizeof(vis)); //顺便把vis初始化
	a[1][1] = 1; //这个要注意，会发现a[1][1]是不符合公式的，所以要单独写
	for (int i = 2; i <= 12; i++) {
		for (int j = 1; j <= i; j++) {
			a[i][j] = a[i - 1][j] + a[i - 1][j - 1];  //这个公式很好推的，多列几项知道了
		}
	}
	//打印杨辉三角代码
	for (int i = 1; i <= 12; i++) {
		for (int j = 1; j <= i; j++) {
			cout << a[i][j] << "  ";
		}
		cout << endl;
	}
}

//这题需要排列，所以用dfs更方便，没有排列我还是习惯用bfs。。。。
//本题其实还可以使用，stl的next_permutation()函数，但会相对慢一点，感兴趣的同学可以自行百度（很好用的函数*_*）
void dfs(int k,int ans) {
	//控制程序退出，其实也可以用exit（0）但我最开始上c语言课时好像听老师说不要随便用exit，不懂，打瞌睡去了/*/&%￥#@
	if (flag == 1) {
		return;
	}
	if (ans > sum) return;  //剪枝，如果已经大于sum就没必要继续深搜了，注意这个剪枝放的位置
	//注意这里是n+1,这取决于你的编码习惯
	if (k == n+1 && ans == sum) {
		for (int i = 1; i <= n; i++) {
			cout << c[i] << " ";
		}
		flag = 1;
		return;
	}
	for (int i = 1; i <= n; i++) {
		if (!vis[i]) {
			vis[i] = true;
			c[k] = i;
			dfs(k + 1, ans + a[n][k] * i);  //添加的每个数字都要乘以它前面的系数（杨辉三角）
			vis[i] = 0;
		}
	}
}

int main() {
	init();
	cin >> n >> sum;
	dfs(1, 0);
	return 0;
}