Query kth smallest trimmed number

September 6, 2022


Problem URL: Query kth smallest trimmed number

We will take the nums, iterate over each query, and take the last trim values of the nums and store it as integer in a values array along with the index. Then we sort this values array and take the kth value and append it to our result. After the iteration of the queries are done, we return the result.

class Solution:
    def smallestTrimmedNumbers(self, nums: List[str], queries: List[List[int]]) -> List[int]:
        res = []
        for k, trim in queries:
            values = [(int(num[-trim:]), i) for i, num in enumerate(nums)]
            _, idx = sorted(values, key=lambda x: x[0])[k-1]
        return res

Time Complexity: O(q*nlog(n)), q is the number of queries, n is the number of elements in nums
Space Complexity: O(n)