P1088 [NOIP2004 普及组] 火星人

题目描述

人类终于登上了火星的土地并且见到了神秘的火星人。人类和火星人都无法理解对方的语言，但是我们的科学家发明了一种用数字交流的方法。这种交流方法是这样的，首先，火星人把一个非常大的数字告诉人类科学家，科学家破解这个数字的含义后，再把一个很小的数字加到这个大数上面，把结果告诉火星人，作为人类的回答。

火星人用一种非常简单的方式来表示数字――掰手指。火星人只有一只手，但这只手上有成千上万的手指，这些手指排成一列，分别编号为 1,2,3,⋯。火星人的任意两根手指都能随意交换位置，他们就是通过这方法计数的。

一个火星人用一个人类的手演示了如何用手指计数。如果把五根手指――拇指、食指、中指、无名指和小指分别编号为 1,2,3,4 和 5，当它们按正常顺序排列时，形成了 5 位数 12345，当你交换无名指和小指的位置时，会形成 5 位数 12354，当你把五个手指的顺序完全颠倒时，会形成 54321，在所有能够形成的 120 个 5 位数中，12345 最小，它表示 1；12354 第二小，它表示 2；54321 最大，它表示 120。下表展示了只有 3 根手指时能够形成的 6 个 3 位数和它们代表的数字：

三进制数	代表的数字
123	1
132	2
213	3
231	4
312	5
321	6

现在你有幸成为了第一个和火星人交流的地球人。一个火星人会让你看他的手指，科学家会告诉你要加上去的很小的数。你的任务是，把火星人用手指表示的数与科学家告诉你的数相加，并根据相加的结果改变火星人手指的排列顺序。输入数据保证这个结果不会超出火星人手指能表示的范围。

输入格式

共三行。
第一行一个正整数 N，表示火星人手指的数目（1≤N≤10000）。
第二行是一个正整数 M，表示要加上去的小整数（1≤M≤100）。
下一行是 1 到 N 这 N 个整数的一个排列，用空格隔开，表示火星人手指的排列顺序。

输出格式

N 个整数，表示改变后的火星人手指的排列顺序。每两个相邻的数中间用一个空格分开，不能有多余的空格。

输入输出样例

输入 #1

5
3
1 2 3 4 5

输出 #1

1 2 4 5 3

思路

这道题也是其实就是全排列问题的演变，如果不清楚的话可以先去参考我之前发过的讲解P1706 全排列问题，然后再来理解这道题，所以我也不具体讲解是怎么进行基本的操作了，我们讲一下，这道题有哪些变化。

这道题先给了我们火星人手指的数目N，以及我们要加上去的数和火星人现在以“火星语”表示的数字。那么我们在深搜的递归中step要满足的就是N,我们还需将火星人的数字与我们自己的数字相加，最后以“火星语”的方式输出。

一开始我的思路是先找出火星人表示的数字是我们正常数字的几，再加上我们自己的，然后再依次递归，每次判断是否为我们相加得到的数字，一直到我们找到为止。

但是后面提交了后发现超时了，后来去学习了一下发现，这个不像平时的深搜我们需要将全部答案都搜索出来，当我们找到我们的目标后，就可以进行剪枝了，即增加一个初始值为false的bool全局变量，当达到目标后，就将此变量变为true，在递归前面加上一个if语句，如果此变量为真，那么return。

那么我们做完了一些修改，是不是就AC了呢，结果提交后还是发现有超时！！！究竟是为什么呢？后面我再重新想了一下，我们为何一定得找出火星人表示的数字是我们正常数字的几，如果它是一个很大很大的数字，那么我们前面就需要递归很多次，但这些都没有什么用处，我们何不直接从火星人表示的数字出发，然后加上一个cnt来记录是火星人表示的数字的后面第几个，当从cnt == M时，就是我们所需要的那个数字啦，然后就直接输出就完事了。最后去试了一下，终于也是AC了。

接下来就是大家最期待的AC代码咯。

AC代码

#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
int n, num, cnt = 0, sum = 0;;
int a[10002],b[10002];
int book[10002];
bool t = true;
void dfs(int temp) {	
	if (t == false) {
		return;//剪枝
	}
	if (temp == n) {
		cnt++;
		if (cnt == num + 1) {
			for (int i = 0; i < n; i++) {
				cout << a[i] << " ";
				t = false;
			}
		}
	}
	for (int i = 1; i <= n; i++) {
		if (!cnt) {
			i = b[temp];//直接从火星人表示的数出发
		}
		if (book[i] != 1) {
			a[temp] = i;
			book[i] = 1;
			dfs(temp + 1);
			book[i] = 0;
			a[temp] = 0;
		}
	}
}
void solve()
{
	cin >> n >> num;
	memset(book, 0, sizeof(book));
	for (int i = 0; i < n; i++) {
		cin >> b[i];
	}
	dfs(0);
}
int main()
{
	int t = 1;
	cin.tie(0)->ios::sync_with_stdio(false);
	while (t--) {
		solve();
	}
	return 0;
}