[LeetCode] 977. Squares of a Sorted Array

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

Squares of a Sorted Array - LeetCode

Leetcode - Squares of a Sorted Array

문제 유형 : 정렬

문제 난이도 : Easy

 

문제

Given an integer array nums sorted in non-decreasing order, return an array of the squares of each number sorted in non-decreasing order.

정수 배열 nums를 오름차순으로 준다. 각 숫자의 제곱수 배열을 오름차순으로 반환하시오.

 

풀이

각 숫자에 대해서 제곱을 한 뒤, 정렬해주면 된다.

 

코드

C++

#pragma GCC optimize("03", "unroll-loops");
static const int __ = [](){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    return 0;
}();

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        for(auto &num : nums) {
            num = num*num;
        }

        sort(nums.begin(), nums.end());
        return nums;
    }
};