leetcode 1387.将整数按权重排序

题目描述：

我们将整数 x 的 权重 定义为按照下述规则将 x 变成 1 所需要的步数：

如果 x 是偶数，那么 x = x / 2
如果 x 是奇数，那么 x = 3 * x + 1
比方说，x=3 的权重为 7 。因为 3 需要 7 步变成 1 （3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1）。

给你三个整数 lo， hi 和 k 。你的任务是将区间 [lo, hi] 之间的整数按照它们的权重 升序排序 ，如果大于等于 2 个整数有 相同 的权重，那么按照数字自身的数值 升序排序 。

请你返回区间 [lo, hi] 之间的整数按权重排序后的第 k 个数。

注意，题目保证对于任意整数 x （lo <= x <= hi） ，它变成 1 所需要的步数是一个 32 位有符号整数。

链接: https://leetcode-cn.com/problems/sort-integers-by-the-power-value/.

解题思路：

定义一个二维数组，第一列表示元素变成1的步数，第二列对应元素的初始值，这样sort排序就可以直接输出[k][1]，求步数就一个while循环就可以。

学习时间：

class Solution {
public:
    int getKth(int lo, int hi, int n) {
        vector<int> temp;
        vector<vector<int>> result;
        for(int k=lo;k<=hi;k++){
            int i=k;
            int step=0;
            while(i!=1){
                if(i%2==0) i=i/2;
                else i=3*i+1;
                step++;
            }
            temp.clear();
            temp.push_back(step);
            temp.push_back(k);
            result.push_back(temp);
        }
        sort(result.begin(),result.end());
        return result[n-1][1];
    }
};