Build Array from Permutation LeetCode solution in Python

1
class Solution:
2
    # As maximum value of the array element is 1000, this solution would work
3
    def buildArray(self, nums: List[int]) -> List[int]:
4
        for i in range(len(nums)):
5
            if nums[nums[i]] <= len(nums):
6
                nums[i] = nums[nums[i]] * 1000 + nums[i]
7
            else:
8
                nums[i] = mod(nums[nums[i]], 1000) * 1000 + nums[i]
9

10
        for i in range(len(nums)):
11
            nums[i] = nums[i] // 1000
12

13
        return nums