大家知道,给出正整数n,则1到n这n个数可以构成n!种排列,把这些排列按照从小到大的顺序(字典顺序)列出,如n=3时,列出1 2 3,1 3 2,2 1 3,2 3 1,3 1 2,3 2 1六个排列。 任务描述: 给出某个排列,求出这个排列的下k个排列,如果遇到最后一个排列,则下一个排列为第1个排列,即排列1 2 3…n。 比如:n=3,k=2给出排列2 3 1,则它的下一个排列为3 1 2,下两个排列为3 2 1,因此答案为3 2 1。
输入 第一行是一个正整数m,表示测试数据的个数,下面是m组测试数据,每组测试数据第一行是两个正整数n(1≤n < 1024)和k(1≤k≤64),第二行有n个正整数,是1 2 … n的一个排列。
输出 对于每组输入数据,输出一行,n个数,中间用空格隔开,表示输入排列的下k个排列。
解法1:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int k,t,i,j,p,n,a[1100];
int main()
{
scanf(%d,&t);
while(t--)
{
scanf("%d",&n);
for(j=0; j<n; j++)
scanf("%d",a[j]);
while(k--)
next_permutation(a,a+n);
for(j=0; j<n-1; j++)
print("%d ",&a[j]);
printf("%d\n",a[n-1]);
}
return 0;
}
直接用了全排列生成算法next_permutation,算出下一个全排列。解法2:
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int a[1050];
int main()
{
int T;
cin>>T;
while(T--)
{
int n,k;
scanf("%d%d",&n,&k);
for(int i=0; i<n; i++)
{
scanf("%d",&a[i]);
}
if(n==1)
{
printf("1\n");
continue;
}
for(;;)//找一个大于a[i],并且在所有大于a[i]的元素中最小的
{
if(k==0)
break;
int i,j;
for(i=n-1; i>=1; i--) //从右往左找一个非递增的位置i
{
if(a[i-1]<a[i])
break;
}
if(i!=0)//从i位置之后寻找大于a[i]的并且在这其中最小的数,一般最小的数就是最后一个比它大的数
{
i--;
for(j=i+1; j<n; j++)
{
if(a[j]<a[i])
break;
}
j--;
swap(a[i],a[j]);//找到大于且最小的那个数之后,交换两个数
sort(a+i+1,a+n);//从第i个元素开始,对后面进行排序
}
else
{
for(int i=0; i<n; i++)
a[i]=i+1;
}
k--;
}
for(int i=0; i<n-1; i++)
printf("%d ",a[i]);
printf("%d\n",a[n-1]);
}
return 0;
}
运行结果: