力扣(LeetCode)756

題目地址:
https://leetcode-cn.com/probl...
題目描述:
給出集合 [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-1次下一個排列。
第二種是回溯法求全排列,設置一個全局變量cur爲當前求出的排列數,求出第k個全排列,也就是cur==k時,停止所有遞歸(否則會超時)。雖然基於交換遞歸的方法也可以求全排列,但是那種方法求的沒有序的關係,所以只能回溯,而不能用遞歸的方法。我這裏採用第二種方法。
java ac代碼:

class Solution {
    int cur = 0;
    String ans = "";
    public String getPermutation(int n, int k) {
        int[]nums = new int[n];
        for(int i = 0;i < n;i++){
            nums[i] = i+1;
        }
        backtrack(0,n,k,nums,new int[n],new boolean[n]);
        return ans;
    }
    
    void backtrack(int i,int n,int k,int[] nums,int[] A,boolean[]flag)
    {
        if(cur == k)
            return;
        if(i == n)
        {
            cur++;
            if(cur == k)
            for(int j = 0;j < A.length;j++)
                ans+=A[j];
            return;
        }
        for(int j = 0;j < nums.length;j++)
            if(!flag[j])
            {
                A[i] = nums[j];
                flag[j] = true;
                backtrack(i+1,n,k,nums,A,flag);
                flag[j] = false;
            }
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章