본문 바로가기

두두의 알고리즘/문제

[정렬] 릿코드 Easy 977 'Squares of a Sorted Array' (Python)

728x90

<문제 링크>

https://leetcode.com/problems/squares-of-a-sorted-array/

 

Squares of a Sorted Array - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com


<문제 풀이>

  1. 리스트의 각 값에 연산을 해야 하므로 벡터연산이 가능한 numpy를 import 함

<코드>

import numpy as np
class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        answer = np.array(nums) ** 2
        answer.sort()
        return list(answer)

 

<고쳐야 할 점>

  • numpy import 구문 외우기