试题 算法训练 数字游戏
资源限制
内存限制:256.0MB C/C++时间限制:1.0s Java时间限制:3.0s Python时间限制:5.0s
问题描述
给定一个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
首先声明,这篇文章参考“醒醒,写bug了!”。虽然题不难,问题在于自己的思维还太过死板,做的题太少了。所以这篇文章是作为一种批注性质的。可能会有错误,请指正,但轻喷!
#include<bits/stdc++.h>
using namespace std;
const int N = 15;
int n, sum;
int q[N], tmp[N];// 存储答案
bool p, use[N];// 判断是否找到答案、对应q查询是否存在
void dfs(int u)// u表示q元素数量
{
if(p) return; // 结果输出则p==true
if(u == n)// q已经装满,判断答案是否成立
{
memcpy(tmp, q, sizeof tmp); // 将q复制到tmp,需要还原所以不改变原数组
// 按题回推结果
for(int i = 0;i < n-1;i ++)
for(int k = 0;k < n;k ++)
{
tmp[k] += tmp[k+1]; // 数组多开了空间,不需要担心越界
}
// 回推的值存储在tmp[0]
if(tmp[0] == sum) // 如果满足条件就输出
{
p = true;
for(int i = 0;i < n;i ++)
{
cout << q[i] << " ";
// 一般多一个空格不会WA,实在不放心可以特判下
// if(i != n-1) cout << " ";
}
//putchar(10);
}
}
for(int i = 1;i <= n;i ++)
{// (题目)为1~N的一个排列,遍历所有元素的排列
if(!use[i]) // use[i] == true则已经取过不能再用
{
use[i] = true;
q[u] = i;// 选中i值
dfs(u+1);// 元素+1
use[i] = false; //回溯,q不用管理,下次依旧从u赋值
}
}
}
int main()
{
cin >> n >> sum;
dfs(0);
return 0;
}