[LeetCode] Squares of a Sorted Array

vector sort하는 법.

sort(벡터이름.begin(), 벡터이름.end());

 

나는 단순히 제곱 해준 후 sort 쓰는 법 생각했는데 most votes에서는 sort안쓰고, 제곱 후 left most, right most 사용해서 비교하면서 새 vector 만들어서 값 넣어줬다. 

 

sort쓸 때의 속도가 input이 많을 때 직접 비교하는 알고리즘 보다 더 빠를지 느릴지 모르겠다.

내가 짠 코드

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int i;
        for(i=0;i<nums.size();i++){
            nums[i] = nums[i]*nums[i];
        }
        sort(nums.begin(),nums.end());
        return nums;
    }
};

Discuss 코드 1

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        vector<int> res(A.size());
        int l = 0, r = A.size() - 1;
        for (int k = A.size() - 1; k >= 0; k--) {
            if (abs(A[r]) > abs(A[l])) res[k] = A[r] * A[r--];
            else res[k] = A[l] * A[l++];
        }
        return res;
    }
};

 

leetcode.com/explore/featured/card/fun-with-arrays/521/introduction/3240/

Explore - LeetCode