总时间限制:
5000ms
内存限制:
65536kB
描述
题目描述:
大家知道,给出正整数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个排列,即排列1 2 3…n。
比如:n = 3,k=2 给出排列2 3 1,则它的下1个排列为3 1 2,下2个排列为3 2 1,因此答案为3 2 1。
输入
第一行是一个正整数m,表示测试数据的个数,下面是m组测试数据,每组测试数据第一行是2个正整数n( 1 <= n < 1024 )和k(1<=k<=64),第二行有n个正整数,是1,2 … n的一个排列。
输出
对于每组输入数据,输出一行,n个数,中间用空格隔开,表示输入排列的下k个排列。
样例输入
3 3 1 2 3 1 3 1 3 2 1 10 2 1 2 3 4 5 6 7 8 9 10
样例输出
3 1 2 1 2 3 1 2 3 4 5 6 7 9 8 10
#include <bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
int a[1030]={0};
void NextPermutation(int n){
int flag=n-1;
int temp;
while(flag!=0&&a[flag]<a[flag-1])//从最后向前查找,直到找到某个位置,这个位置比前一个位置大
flag--;
if(flag==0){//表明已经是降序,则下一个排列为1~n升序
for(int i=0;i<n;i++)
a[i]=i+1;
return;
}
for(int i=n-1;i>=flag;i--){//在a[flag]~a[n-1]中找到最小的大于a[flag-1]的数
if(a[i]>a[flag-1]){
temp=a[flag-1];
a[flag-1]=a[n-1];
a[n-1]=temp;
//printf("***%d***",i);
break;//因为从左往右,找到的第一个大于a[flag]的数一定是大于a[flag]数中最小的
}
}
while(n-1>flag){//将位置flag~n的所有数从小到大排列
temp=a[n-1];
a[n-1]=a[flag];
a[flag]=temp;
flag++;
n--;
}
}
int main(){
int t,n,k;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
for(int i=0;i<k;i++)
NextPermutation(n);
for(int i=0;i<n-1;i++)
printf("%d ",a[i]);
printf("%d\n",a[n-1]);
}
return 0;
}
还有一种超简单代码(利用c++中的next_permutation):
#include <bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
int a[1030]={0};
int main(){
int t,n,k;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
for(int i=0;i<k;i++)
next_permutation(a,a+n);
for(int i=0;i<n;i++)
printf("%d ",a[i]);
printf("\n");
}
return 0;
}