[LeetCode 1387] 将整数按权重排序
题目描述
题目链接:[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 位有符号整数。
示例1
输入:lo = 12, hi = 15, k = 2
输出:13
解释:12 的权重为 9(12 --> 6 --> 3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1)
13 的权重为 9
14 的权重为 17
15 的权重为 17
区间内的数按权重排序以后的结果为 [12,13,14,15] 。对于 k = 2 ,答案是第二个整数也就是 13 。
注意,12 和 13 有相同的权重,所以我们按照它们本身升序排序。14 和 15 同理。
示例2
输入:lo = 7, hi = 11, k = 4
输出:7
解释:区间内整数 [7, 8, 9, 10, 11] 对应的权重为 [16, 3, 19, 6, 14] 。
按权重排序后得到的结果为 [8, 10, 11, 7, 9] 。
排序后数组中第 4 个数字为 7 。
提示
- 1 <= lo <= hi <= 1000
- 1 <= k <= hi - lo + 1
思路分析
1.题目的数量级并不是很大,可以看出最多只有1000个数,直接暴力即可
2.用pair存储数字以及该数字的权重,需要重写cmp函数,因为默认是按照pair.first排序的,而这里需要根据数字的权重排序
3.cmp函数需要使用static修饰,不然会报错
代码
class Solution {
public:static bool cmp(pair<int, int> a, pair<int, int> b){if(a.second == b.second) return a.first < b.first;else return a.second < b.second;}int func(int n){int res = 0;while(n != 1){if(n % 2) n = 3 * n + 1;else n /= 2;res++;}return res;}int getKth(int lo, int hi, int k) {vector<pair<int, int>> v;for(int i = lo; i <= hi; i++) v.push_back({i, func(i)});sort(v.begin(), v.end(), cmp);return v[k - 1].first;}
};
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!
