[Leet Code] 1920. Build Array from Permutation

문제 원본: https://leetcode.com/problems/build-array-from-permutation/

문제

0부터 시작하는 nums 배열이 주어졌을 때, 같은 길이를 갖고 ans[i] = nums[nums[i]]인 ans 배열을 만들어 반환하라. i는 0 이상 nums.length 미만이다.

0부터 시작하는 배열 nums는 0부터 nums.length - 1까지의 각각 다른 정수로 이루어진 배열이다.

Example 1:

Input: nums = [0,2,1,5,3,4]
Output: [0,1,2,4,5,3]
Explanation: The array ans is built as follows: 
ans = [nums[nums[0]], nums[nums[1]], nums[nums[2]], nums[nums[3]], nums[nums[4]], nums[nums[5]]]
    = [nums[0], nums[2], nums[1], nums[5], nums[3], nums[4]]
    = [0,1,2,4,5,3]

Example 2:

Input: nums = [5,0,1,2,3,4]
Output: [4,5,0,1,2,3]
Explanation: The array ans is built as follows:
ans = [nums[nums[0]], nums[nums[1]], nums[nums[2]], nums[nums[3]], nums[nums[4]], nums[nums[5]]]
    = [nums[5], nums[0], nums[1], nums[2], nums[3], nums[4]]
    = [4,5,0,1,2,3]

Constraints:

• 1 <= nums.length <= 1000
• 0 <= nums[i] < nums.length
• nums의 요소는 서로 다르다.

접근 방법

말 그대로 반복문을 돌면서 ans[i] = nums[nums[i]]를 0부터 nums.length 전까지만 실행해주면 된다.

풀이

Java

class Solution {
    public int[] buildArray(int[] nums) {
        int[] ans = new int[nums.length];
        
        for (int i = 0; i < nums.length; i++) {
            ans[i] = nums[nums[i]];
        }
        return ans;
    }
}

• Runtime: 2ms
• Memory: 54.1MB

C#

public class Solution {
    public int[] BuildArray(int[] nums) {
        int[] ans = new int[nums.Length];
        
        for (int i = 0; i < nums.Length; i++)
            ans[i] = nums[nums[i]];
        return ans;
    }
}

• Runtime: 219ms
• Memory: 45.2MB

Python

class Solution:
    def buildArray(self, nums: List[int]) -> List[int]:
        ans = [nums[nums[i]] for i in range(len(nums))]
        return ans

• Runtime: 158ms
• Memory: 14.1MB