LeetCode-Permutation Sequence-计数搜索
https://oj.leetcode.com/problems/permutation-sequence/
一道简单题目。但是要一次写出bug-free的代码还是挺困难。
子问题是固定了前i个数,对第i+1数,要求从他开始往后的第k个数即可。第i+1个数由以下两个因素确定:
1)[i+2,n)这些数字共有多少个数?
2)还剩下哪些数字?
条件1)能够决定这个数字在余下数字里排第几。条件2)决定了这个数具体是啥。
需要注意题目里的k是从1开始数。最好先把它转成从0开始索引。而后通过条件2确定这个数时也要小心的数到这个数具体是几。
class Solution { public: string getPermutation(int n, int k) { string s(n,'0'); vector <bool> nums(n,false); int A=1; for (int i=1;i<=n;i++) A*=i; k--; for (int i=0;i<n;i++){ A/=(n-i); int cur=k/A; k=k%A; int p=0; while(cur>=0){ if (!nums[p]) cur--; p++; } p--; s[i]='0'+p+1; nums[p]=true; } return s; } };