[leetCode] 1920. Build Array from Permutation (Python)

▼ 문제 바로가기 (링크) ▼ 

https://leetcode.com/problems/build-array-from-permutation/

Build Array from Permutation - LeetCode

---

*Permutation: 순열 (: 고정된 개수를 가진 수들을 일정 규칙에 따라 순서대로 나열한 수열)

 

수열 nums 가 input으로 주어진다.

인덱스 값을 다시 인덱스로 사용해서 만든 새로운 순열을 반환하는 문제.

---

#최초 코드 (기본적인 for문)
class Solution:
    def buildArray(self, nums: List[int]) -> List[int]:
        ans = []
        for i in range(0, len(nums)):
            ans.append(nums[nums[i]])
        return ans
        
# 컴프리헨션 ( 훨씬 빠름)
class Solution:
    def buildArray(self, nums: List[int]) -> List[int]:
	    return [nums[i] for i in nums]

처음은 기본 for문으로 구현하고, 이후 컴프리헨션으로 개선했다.

실행 속도가 확실히 빠르다.