Leetcode60 Permutation Sequence

题目

这道题目是来自leetcode的,第60题:题目的网址

题目的截图

思路

这道题目意思是输入一个n,1-n任意顺序呢可以组成一个数,从小到大排序这些数,找到第k个大小的数。

这题我拿到手先搞了个递归的方法,就是把所有的数都求出来,选第k个,超时了绝大多数的案例,然后我优化了一下递归,在取到第k个值的时候返回,多通过了一些测试案例。感觉递归这条路走不下去了。后来我看了一篇CSDN的博客,博客讲的是康拓编码,感觉很高端,我大致知道个思路,准备自己写,我觉得用set更加好,set不会存放重复的数据,而且还可以erase掉指定的value,还可以排序,这个很重要。下面就介绍一下这题最终的思路。

首先将初始化一个栈,将0 - n-1的阶乘放入栈中,然后初始化一个set,将1-n放入set中,初始化完毕以后,你就可以大展身手啦,每次从栈中弹出一个阶乘,这个阶乘一定是n-1位的所有的可能性的值的个数。如下图,n = 4,3! 为6,那么我们现在用k/6得到的一定是第几个大组,这里k是9,这里需要k--,方便计算求余。8 / 6为1,相当于第二组,那么就是从我们的set中将第二个取出来,取出的是2,将set中的2 erase掉,重新自动排序,重复刚刚的过程。每次取出的值就是set中的第几个。逐个除以并且求余。最后一个直接加入。

样例截图

代码

//
//  main.cpp
//  leetcode60
//
//  Created by 李林 on 2017/1/13.
//  Copyright © 2017年 lee. All rights reserved.
//

#include <iostream>
#include <string>
#include <set>
#include <stack>
using namespace std;

class Solution {
public:
    stack<int> initStack(int n){
        int sum = 1;
        stack<int> st;
        
        for(int i=1; i<n; i++){
            sum = sum * i;
            st.push(sum);
        }
        
        return st;
    }
    
    string getPermutation(int n, int k) {
        string result;
        set<int> nums;
        stack<int> st;
        if(k == 0)  return result;
        
        // 1.初始化
        for(int i=1; i<=n; i++)
            nums.insert(i);
        st = initStack(n);
        
        // 2.计算
        k--;
        while(!st.empty()) {
            int index = k / st.top();
            k %= st.top();
            
            // 2.1取值
            int count = -1;
            for(set<int>::iterator it = nums.begin(); it!=nums.end(); it++){
                count++;
                if(index == count){
                    result.push_back('0' + (*it));
                    nums.erase(*it);
                    break;
                }
            }

            st.pop();
        }
        
        result.push_back('0' + (*nums.begin()));
        
        return result;
    }
};

int main(int argc, const char * argv[]) {
    Solution s;
    
    string result = s.getPermutation(4, 9);
    cout<<result<<endl;
    
    return 0;
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,769评论 0 33
  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 17,920评论 2 36
  • 即“做生意的完整周期=平均销货天数+平均收现天数”。 按照付款周期来理解,以付款给供应商到收到客户货款来分“做生意...
    后亮笔记阅读 684评论 0 0
  • 周三 我作为一枚“凉粉” 去参加了梁冬见面会。 听过梁某人和徐文兵的《黄帝内经》 在听梁某人和吴伯凡的《冬吴同学会...
    陈颖_乐嘉妈妈阅读 206评论 0 3
  • 说起民谣大家应该都知道一些,民谣是文艺青年的标配,而最近几年最火的应该就是《董小姐》,所以先说说宋冬野,一个地道的...
    罅隙阅读 1,256评论 19 12