Sort Array of 0s and 1s

function sortZeroOne(nums) {
  let count0 = 0;
  for (let num of nums) if (num === 0) count0++;
  for (let i = 0; i < nums.length; i++) {
    nums[i] = i < count0 ? 0 : 1;
  }
  return nums;
}
                  

Explanation: We count the number of zeros and overwrite the array accordingly.

⏱ O(n) | 💾 O(1)