[LeetCode] 905. Sort Array By Parity

https://leetcode.com/problems/sort-array-by-parity/description/

LeetCode - The World's Leading Online Programming Learning Platform

문제 유형 : 정렬

문제 난이도 : Easy

 

문제

Given an integer array nums, move all the even integers at the beginning of the array followed by all the odd integers.

Return any array that satisfies this condition.

 

짝수가 배열의 앞, 홀수가 배열의 뒤쪽에 오도록 하라.

 

풀이

람다 함수를 정의해서, 두 수중 짝수인 수를 앞으로 하고, 같은 짝수끼리는 더 작은수를 앞으로 해주기로 했다.

 

코드

C++

class Solution {
public:
    vector<int> sortArrayByParity(vector<int>& nums) {
        sort(nums.begin(), nums.end(), [](const int &a, const int &b) {

            if (a % 2 == 0 && b % 2 == 0)
                return a < b;
            else if (a % 2 != 0 && b % 2 != 0)
                return a > b;
            else if (a % 2 == 0)
                return true;
            else
                return false;
        });

        return nums;
    }
};