Sort the Jumbled Numbers Python Solution

lcpy gen -n 2191   # by problem number
lcpy gen -s sort_the_jumbled_numbers   # by problem name

Input: mapping = [8,9,4,0,2,1,3,5,7,6], nums = [991,338,38]
Output: [338,38,991]
Explanation:
Map the number 991 as follows:
1. mapping[9] = 6, so all occurrences of the digit 9 will become 6.
2. mapping[1] = 9, so all occurrences of the digit 1 will become 9.
Therefore, the mapped value of 991 is 669.
338 maps to 007, or 7 after removing the leading zeros.
38 maps to 07, which is also 7 after removing leading zeros.
Since 338 and 38 share the same mapped value, they should remain in the same relative order, so 338 comes before 38.
Thus, the sorted array is [338,38,991].

Input: mapping = [0,1,2,3,4,5,6,7,8,9], nums = [789,456,123]
Output: [123,456,789]
Explanation: 789 maps to 789, 456 maps to 456, and 123 maps to 123. Thus, the sorted array is [123,456,789].

class Solution:
    # Time: O(n log n * d) where d is the digit count
    # Space: O(n)
    def sort_jumbled(self, mapping: list[int], nums: list[int]) -> list[int]:
        return sorted(nums, key=lambda num: int("".join(str(mapping[int(d)]) for d in str(num))))