查找第k个排列java实现

题目描述

https://leetcode-cn.com/problems/permutation-sequence

给出集合 [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"

思路：一般思路是依次求所有排列，当遍历到第k个时停止，输出当前排列，但这种DFS+回溯的方法会超时...怎么优化剪枝还是依旧超时...

class Solution {
    List<String> res=new ArrayList<String>();
    boolean[] num=null;
    public String getPermutation(int n, int k) {
        String[] s=new String[n];
        for(int i=1;i<=n;i++){
            s[i-1]=String.valueOf(i);
        }
        int x=(k-1)/(jiecheng(n-1));//求当前是第几组排列（以数字几开头的）
        int y=(k-1)%(jiecheng(n-1));//求是这组中的第几个
        String[] s1=new String[n];
        s1[0]=s[x];//找到开头后把它放到新数组的第一个
        for(int i=x+1;i<n;i++){
            s1[i]=s[i];
        }
        for(int i=1;i<=x;i++){
            s1[i]=s[i-1];
        }
        //将剩下的数字从大到小排列
        //如原数组是1 2 3 4 5 6，我们找到是第四组，所以开头是4，数组变为： 4 1 2 3 5 6
        num=new boolean[n];
        String str="";
        int flag=0;
        dfs(s1,str,0,res,num,flag,y+1);//dfs回溯求当前组的第y个即可
        return res.get(y);
    }
    public int jiecheng(int n){//阶乘计算函数
        int result=1;
        for(int i=1;i<=n;i++){
            result=result*i;
        }
        return result;
    }
    public void dfs(String[] s,String str,int start,List<String> res,boolean[] num,int flag,int k){
        if(start==s.length){//当求满n个数后压入res
            res.add(str);
            if(res.size()==k){//求得第k个，停止
                flag=1;
            }
            return;
        }
        if(flag==1){//如果求到第k个就停止
            return;
        }
        for(int i=0;i<s.length;i++){
            if(!num[i]){//若当前数字没用过则执行
                str=str+s[i];
                num[i]=true;//当前数组用过，设为true
                dfs(s,str,start+1,res,num,flag,k);//dfs
                if(flag==1){
                    return;
                }
                str=str.substring(0,str.length()-1);//回溯str
                num[i]=false;//回溯当前数组使用状态
            }
        }
        return;
    }
}

 不超时解法：从数学角度思考
        因为数字都是从1开始的连续自然数, 排列出现的次序可以推
        算出来, 对于n=4, k=15 找到k=15排列的过程:
        
        1 + 对2,3,4的全排列 (3!个)         
        2 + 对1,3,4的全排列 (3!个)         3, 1 + 对2,4的全排列(2!个)
        3 + 对1,2,4的全排列 (3!个)-------> 3, 2 + 对1,4的全排列(2!个)-------> 3, 2, 1 + 对4的全排列(1!个)-------> 3214
        4 + 对1,2,3的全排列 (3!个)         3, 4 + 对1,2的全排列(2!个)         3, 2, 4 + 对1的全排列(1!个)
        
        确定第一位:
            k = 14(从0开始计数)
            index = k / (n-1)! = 2, 说明第15个数的第一位是3 
            更新k
            k = k - index*(n-1)! = 2
        确定第二位:
            k = 2
            index = k / (n-2)! = 1, 说明第15个数的第二位是2
            更新k
            k = k - index*(n-2)! = 0
        确定第三位:
            k = 0
            index = k / (n-3)! = 0, 说明第15个数的第三位是1
            更新k
            k = k - index*(n-3)! = 0
        确定第四位:
            k = 0
            index = k / (n-4)! = 0, 说明第15个数的第四位是4
        最终确定n=4时第15个数为3214 
        **/

class Solution {
    public String getPermutation(int n, int k) {
        if(n==0)
        return "";
        List<String> list=new ArrayList<String>();//用List存放数组，方便查找第k个和删去某个
        for(int i=1;i<=n;i++){
            list.add(String.valueOf(i));
        }
        k=k-1;
        int index=0;
        String res="";
        for(int i=n-1;i>=0;i--){
            index=k/jiecheng(i);
            res=res+list.get(index);
            list.remove(index);
            k=k-jiecheng(i)*index;
        }
        return res;
    }
    public int jiecheng(int k){//阶乘函数
        int res=1;
        for(int i=k;i>0;i--){
            res=res*i;
        }
        return res;
    }
}