P1088 [NOIP 2004 普及组] 火星人c/c++

P1088 [NOIP 2004 普及组] 火星人

题目描述

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

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

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

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

输入格式

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

输出格式

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

输入输出样例

输入

5
3
1 2 3 4 5

输出

1 2 4 5 3

说明/提示

对于 30% 的数据，N≤15。

对于 60% 的数据，N≤50。

对于 100% 的数据，N≤10000。

代码：

#include <bits/stdc++.h>

using namespace std;

vector<int> arr;  // 记录当前方案
vector<int> mars; // 记录火星人的初始排列
vector<bool> st;  // 记录数字是否已经被选过
int nn, mm;
int res = 0;
bool flag=false;
// 深度优先搜索
void dfs(int x) {
	    	if(flag){
	    		    		return ;
			}

    // 如果前nn个数字已经选好了，检查是否满足条件
    if (x > nn) {

        res++;  // 找到一个方案
        if (res == mm + 1) {  // 如果是第mm+1个方案，输出
            for (int i = 1; i <= nn; i++) {
                cout << arr[i] << " ";
            }
            flag=true;
            //cout << endl;
        }
        return;
    }
    
    // 遍历所有数字
    for (int i = 1; i <= nn; i++) {
    	if(!res)
    		i=mars[x];
        if (!st[i]) {  // 如果数字i没有被选过
            st[i] = true;    // 标记为已选
            arr[x] = i;      // 选择数字i
            dfs(x + 1);      // 继续递归
            st[i] = false;   // 回溯，撤销选择
        }
    }
}

int main() {
    cin >> nn;  // 输入火星人手指的数目
    cin >> mm;  // 输入要加上的数字（即要查找的第mm个排列）

    mars.resize(nn + 1); // 初始化mars数组大小
    arr.resize(nn + 1);  // 初始化arr数组大小
    st.resize(nn + 1, false); // 初始化st数组大小，默认false

    // 输入火星人手指的初始排列
    for (int i = 1; i <= nn; i++) {
        cin >> mars[i];
    }

    // 深度优先搜索从第一个数字开始
    dfs(1);

    return 0;
}

代码详细解析

1. 全局变量声明

vector<int> arr;  // 记录当前方案
vector<int> mars; // 记录火星人的初始排列
vector<bool> st;  // 记录数字是否已经被选过
int nn, mm;
int res = 0; // 用来记录已经找到的排列个数
bool flag = false; // 用来标记是否找到了第 mm 个排列，防止继续搜索

• arr: 记录当前深度优先搜索（DFS）生成的排列。
• mars: 记录火星人给定的初始排列，排列中的数字表示手指的顺序。
• st: 一个布尔数组，记录每个数字是否已经被选择过，避免在排列中重复选择相同的数字。
• nn: 火星人手指的数量，即排列的长度。
• mm: 要找的第 mm 个排列的序号。
• res: 记录当前已生成的排列个数。
• flag: 标记是否找到了第 mm 个排列，一旦找到，flag 被设为 true，并停止搜索。

2. 深度优先搜索（DFS）

void dfs(int x) {
    if (flag) {
        return;  // 如果已经找到第mm个排列，退出递归
    }

    // 如果前nn个数字已经选好了，检查是否满足条件
    if (x > nn) {
        res++;  // 找到一个方案
        if (res == mm + 1) {  // 如果是第mm+1个方案，输出
            for (int i = 1; i <= nn; i++) {
                cout << arr[i] << " ";
            }
            flag = true; // 找到目标排列，设置flag为true，停止递归
        }
        return;
    }
    
    // 遍历所有数字
    for (int i = 1; i <= nn; i++) {
    	if(!res)
    		i=mars[x];
        if (!st[i]) {  // 如果数字i没有被选过
            st[i] = true;    // 标记为已选
            arr[x] = i;      // 选择数字i，放入当前位置
            dfs(x + 1);      // 继续递归，寻找下一个数字
            st[i] = false;   // 回溯，撤销选择
        }
    }
}

DFS函数的工作流程：

• x 是当前递归的深度，表示当前排列的第 x 个位置。
• 基准条件：
  • 如果 flag 为 true，说明已经找到了第 mm 个排列，直接返回，停止递归。
  • 如果 x > nn，表示已经选好了前 nn 个数字，也就是一组完整的排列。在这种情况下，检查当前的排列是否是第 mm 个。如果是，就打印该排列并设置 flag = true，表示已经找到目标排列。
• 遍历所有数字：
  • 在 for 循环中，我们遍历 1 到 nn 的所有数字。如果当前数字 i 没有被选过（即 st[i] == false），就选择它，放入当前的 arr[x] 中。
  • 然后递归调用 dfs(x + 1)，继续处理下一个位置。
  • 在递归结束后，执行回溯操作：将 arr[x] 置为 0，并将 st[i] 置为 false，表示撤销选择，回到上一层递归。

3. 主函数

int main() {
    cin >> nn;  // 输入火星人手指的数目
    cin >> mm;  // 输入要加上的数字（即要查找的第mm个排列）

    mars.resize(nn + 1); // 初始化mars数组大小
    arr.resize(nn + 1);  // 初始化arr数组大小
    st.resize(nn + 1, false); // 初始化st数组大小，默认false

    // 输入火星人手指的初始排列
    for (int i = 1; i <= nn; i++) {
        cin >> mars[i];
    }

    // 深度优先搜索从第一个数字开始
    dfs(1);

    return 0;
}

主函数的工作流程：

• 首先输入火星人手指的数量 nn 和要查找的排列序号 mm。
• mars.resize(nn + 1)：调整 mars 数组的大小，确保它有 nn + 1 个元素。
• arr.resize(nn + 1)：调整 arr 数组的大小，用来存储当前的排列。
• st.resize(nn + 1, false)：初始化 st 数组，用来记录每个数字是否已经被选过，初始值为 false。
• 输入火星人手指的初始排列，存储到 mars 数组中。
• 调用 dfs(1) 开始递归深度优先搜索，从排列的第一个位置开始。

注意事项

1. if (!res) 的判断：在 DFS 中，res 是一个计数器，表示已找到的排列数。
2. i = mars[x]：i是数字，这里直接将数字i等于当前初试位置的mars【x】值，后面自增，确保了后面的序列比前面的序列大。
3. flag 的使用：flag 用来标记是否已经找到第 mm 个排列。它确保了找到目标排列后，停止所有递归，避免不必要的计算。

总结

• 深度优先搜索：我们通过 DFS 来生成所有排列，并在找到第 mm 个排列时立即输出结果。
• 回溯：回溯操作保证了每个数字在排列中只被使用一次。
• 优化：通过 flag 来控制是否停止递归，确保算法效率。