将整数按权重排序

/*
 * 思路：
 * 1. 使用Stream API来处理区间 [lo, hi] 的整数。
 * 2. 计算每个整数的权重，并将整数和权重映射为数组。
 * 3. 按照权重和数值进行排序。
 * 4. 提取排序后的第 k 个元素。
 */

import java.util.Comparator;
import java.util.stream.IntStream;

public class Solution {
    public int getWeight(int x) {
        int steps = 0;
        while (x != 1) {
            if (x % 2 == 0) {
                x = x / 2;
            } else {
                x = 3 * x + 1;
            }
            steps++;
        }
        return steps;
    }

    public int getKth(int lo, int hi, int k) {
        return IntStream.rangeClosed(lo, hi)
            .mapToObj(i -> new int[]{i, getWeight(i)})
            .sorted(Comparator.<int[]>comparingInt(a -> a[1]).thenComparingInt(a -> a[0]))
            .mapToInt(a -> a[0])
            .skip(k - 1)
            .findFirst()
            .orElse(-1);
    }
}