[자바 JAVA/알고리즘] LeetCode : Squares of a Sorted Array

GitHub https://github.com/vellimole0621

 

LeetCode : Squares of a Sorted Array

 

(문제 - https://leetcode.com/explore/learn/card/fun-with-arrays/523/conclusion/3574/)

 

문제 설명 :   정수로 이루어진 배열에서 각 요소를 제곱해 오름차순으로 정렬하라.

 

풀이 :

각각의 원소를 제곱(제곱하는 경우 마이너스인 요소 문제도 자동 해결된다.) -> Arrays.sort 메소드 사용해서 정렬 -> 결과 return

 

// leetcode : Squares of a Sorted Array
// https://leetcode.com/explore/learn/card/fun-with-arrays/523/conclusion/3574/

import java.util.Arrays;

class Solution {
    public int[] sortedSquares(int[] nums) {

        // 각각의 원소를 제곱함(제곱하면 마이너스인 요소도 자동 해결)
        for(int i = 0; i < nums.length; i++){
            if(nums[i] != 0){
                nums[i] = nums[i] * nums[i];
            }
        }

        // 원소 정렬
        Arrays.sort(nums);

        // 결과 반환
        return nums;
    }
}