LeetCode-60.第k个排列(medium):
给出集合 [1,2,3,…,n],其所有元素共有 n! 种排列。
按大小顺序列出所有排列情况,并一一标记,当 n = 3 时, 所有排列如下:
"123"
"132"
"213"
"231"
"312"
"321"
给定 n 和 k,返回第 k 个排列。
说明:
给定 n 的范围是 [1, 9]。
给定 k 的范围是[1, n!]。
示例 1:
输入: n = 3, k = 3
输出: "213"
示例 2:
输入: n = 4, k = 9
输出: "2314"
我的解法:
说明:不知道是不是因为国庆网络安全问题,LeetCode英文版用不了了,没关系,以后改用中文。。。
这道题我一看,这不是递归问题吗,那就写一个完美的递归程序吧。
class Solution {
public String getPermutation(int n, int k) {
int[] flag = new int[n];
Arrays.fill(flag, 0);
List<String> result = Recur(n, flag);
String ans = "";
if(result.size() == factor(n) && k<=factor(n)) {
ans = result.get(k-1);
}
return ans;
}
public List<String> Recur(int n, int[] flag) {
List<String> ans = new ArrayList<String>();
for (int i = 0; i < flag.length; i++) {
if(flag[i] == 0) {
char head = (char)(i+1+'0');
flag[i] = 1;
if(n == 1) {
String tmp1 = "" + head;
ans.add(tmp1);
flag[i] = 0;
return ans;
}
List<String> tmp = Recur(n-1, flag);
for (int j = 0; j < tmp.size(); j++) {
tmp.set(j, head + tmp.get(j));
ans.add(tmp.get(j));
}
flag[i] = 0;
}
}
return ans;
}
//阶乘函数
public int factor(int t) {
int num = 1;
while(t!=0) {
num*=t;
t--;
}
return num;
}
}
程序就是典型的递归,用flag数组标记用过的数字,用n的大小限定递归的深度。结果,超时了。。。百思不得其解,看了网上的解答发现,这道题根本不是考察递归算法的,明明是脑筋急转弯。。。
大神解法:
public class Solution {
public String getPermutation(int n, int k) {
int[] flag = new int[n];
String result ;
char[] ans = new char[n];
int count = 1;
for( int i = 2;i<n;i++)
count*=i;
for( int i = 0;i<n;i++){
if ( k == 1){
for( int j = 0;j<n;j++){
if( flag[j] == 0){
ans[i] = (char) (j+49);
i++;
flag[j] = 1;
}
}
return new String(ans);
}
int num = (k-1)/count;
k = k - num*count;
count = count/(n-1-i);
for( int j = 0;j<n;j++){
if( flag[j] == 0){
if( num == 0){
ans[i] = (char) (j+49);
flag[j] = 1;
break;
}else
num--;
}
}
}
return new String(ans);
}
}
大体思路就是逐位判断,例如n等于4的情况,第一位为1,2,3,4其中一个,每一个对应(n-1)!=6种情况,然后根据题目要求返回的位置缩小范围,例如k=12,则第一位必为2。第二位和第一位的判断方法类似。