LeetCode in Kotlin

3467. Transform Array by Parity

Easy

You are given an integer array nums. Transform nums by performing the following operations in the exact order specified:

  1. Replace each even number with 0.
  2. Replace each odd numbers with 1.
  3. Sort the modified array in non-decreasing order.

Return the resulting array after performing these operations.

Example 1:

Input: nums = [4,3,2,1]

Output: [0,0,1,1]

Explanation:

Example 2:

Input: nums = [1,5,1,4,2]

Output: [0,0,1,1,1]

Explanation:

Constraints:

Solution

class Solution {
    fun transformArray(nums: IntArray): IntArray {
        val size = nums.size
        val ans = IntArray(size)
        var countEven = 0
        for (i in nums.indices) {
            if (nums[i] and 1 == 0) {
                countEven++
            }
        }
        for (i in countEven until size) {
            ans[i] = 1
        }
        return ans
    }
}