蓝桥杯学习记录||ALGO-1005 数字游戏

数字游戏

问题描述
　给定一个1～N的排列a[i]，每次将相邻两个数相加，得到新序列，再对新序列重复这样的操作，显然每次得到的序列都比上一次的序列长度少1，最终只剩一个数字。
　　例如:
　　3 1 2 4
　　4 3 6
　　7 9
　　16
　　现在如果知道N和最后得到的数字sum，请求出最初序列a[i]，为1～N的一个排列。若有多种答案，则输出字典序最小的那一个。数据保证有解。
输入格式

第1行为两个正整数n，sum

样例格式

一个1～N的一个排列

样例输入

4 16

样例输出

3 1 2 4

数据规模和约定

0<n<=10

解析：本题难度主要在于全排序，这个实现起来有点困难，可以借用next_permutation(全排列算法) 就会简单很多。

next_permutation不懂可以移步这位博主讲的

由图可知 当N等于4时 a被加了1 b被加了3次 …
左边那个数列其实是杨辉三角形。 这样库快速求出sum
图源:Lyz.

代码思路 ：先对 1 ~ n 的数据存于a[]中，然后对a进行全排序，如果在排序中求出了sum就输出a[].

#include <iostream>
#include <algorithm>
using namespace std;
int a[11], n, sum;
int b[11][11];
void cal() //求一下杨辉三角形 
{
	b[1][1]=1;
	for(int i = 2; i <= 10; i ++)
	{
		for(int j = 1; j <= i; j ++)
		{
			b[i][j] = b[i - 1][j] + b[i - 1][j - 1];
		}
	}
}
int main()
{
    cin >> n >> sum;
    
    for (int i = 1; i <= n; i++) a[i] = i;
    cal();
    do
    {
		int s = 0;
		for(int i = 1; i <= n; i ++)
		{
			s += a[i] * b[n][i];
		} 
		if(s == sum)
		{
			for(int i = 1; i <= n; i ++)
			{
				printf("%d ",a[i]);
			}
			break;
		}
    } while (next_permutation(a + 1, a + n + 1));
}

本题:数字游戏