显而易见,只是一道全排列的题目,有了P1076的经验这道题应该不算难,dfs方法。
这道题重点是起始的全排列被给了出来,需要往后遍历出比它大的。
本题一共涉及4个变量和3个数组:
n:题目中火星人的手指数量
m:需要向后推的数
flag:用来记录比火星人给出数大的数的数量
x:操作的数的位置
mars数组:记录火星人给的数
sta数组:由于每个数字只能选择一次,记录各个数字的状态
ans数组:最终答案
这道题一共有个2个需要解决的问题:
1.如何从火星人给出的数字开始向后全排列
2.flag为多少时应该输出答案
大致思路:
从第一位数开始调用dfs,如果flag为0,证明刚开始进行搜索,让这一位数字等于火星人给出的同位数字,sta数组记录数字状态,使用过的数字记为true,继续向下搜索x+1位,如果下x>n说明已经将所有的数字操作完,如果符合条件输出,不符合return掉继续搜索,然后回溯一下。
为了防止输出结果后程序仍然进行一些不必要的搜索,输出后直接退出程序。
具体代码如下
#include "iostream"
using namespace std;
int mars[10005],n,m,ans[10005],flag;
bool sta[10005];
void dfs(int x){
if(x>n){
flag++;
if(flag==m+1){//火星人给出的排列为1,所以要加上1
for(int i=1;i<=n;i++)cout<<ans[i]<<" ";
exit(0);//输出结果退出程序
}
return;
}
for(int i=1;i<=n;i++){
if(!flag)i=mars[x];//让i从火星人给出的数字为起始开始搜索
if(!sta[i]){//如果i数字没有被选中,让x位等于i,并标记i的状态
ans[x]=i;
sta[i]=true;
dfs(x+1);//继续搜索下一位
ans[x]=0;//状态回溯
sta[i]=false;
}
}
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>mars[i];
dfs(1);
}