From ae6414896d53444027a5406c5a8febca35e67fd0 Mon Sep 17 00:00:00 2001 From: bumpsoo Date: Wed, 24 Jul 2024 21:25:40 +0900 Subject: [PATCH] https://leetcode.com/problems/sort-the-jumbled-numbers --- medium/sort_the_jumbled_numbers.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 medium/sort_the_jumbled_numbers.py diff --git a/medium/sort_the_jumbled_numbers.py b/medium/sort_the_jumbled_numbers.py new file mode 100644 index 0000000..c35a26b --- /dev/null +++ b/medium/sort_the_jumbled_numbers.py @@ -0,0 +1,18 @@ +# https://leetcode.com/problems/sort-the-jumbled-numbers +from typing import List + +class Solution: + def sortJumbled(self, mapping: List[int], nums: List[int]) -> List[int]: + def convert(val: int) -> int: + ret = 0 + base = 1 + while val >= 10: + ret += mapping[val % 10] * base + base *= 10 + val //= 10 + ret += mapping[val % 10] * base + return ret + return [ + nums[i] for i in sorted(range(len(nums)), + key=lambda i: (convert(nums[i]), i)) + ]