Leetcode 1488 Solution

This article provides solution to leetcode question 1488 (sort-integers-by-the-power-value)
Categories:
Tags:
Access this page by simply typing in "lcs 1488" in your browser address bar if you have bunnylol configured.

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

Solution

class Solution:
    def getKth(self, lo: int, hi: int, k: int) -> int:
        memo = {}
        def get_power(v):
            nonlocal memo

            if v == 1:
                return 0

            if v in memo:
                return memo[v]

            if v % 2 == 0:
                memo[v] = get_power(v // 2) + 1
            else:
                memo[v] = get_power(3 * v + 1) + 1

            return memo[v]

        data = [(get_power(i), i) for i in range(lo, hi + 1)]
        data.sort()

        return data[k - 1][1]