c++ 코딩테스트 array에서 0을 뒤로 보내기(Move zeros)

문제 {1,3,0,0,0,4,0,5}를 {1,3,4,5,0,0,0,0}으로 변환하라

 

코드 >>

#include <iostream>
#include <array>
#include <vector>
#include <algorithm>
 
// 0 움직이기
// [1,3,4,0,0,4,0,5]
 
void swap(int& a, int& b)
{
    int tmp = a;
    a = b;
    b = tmp;
}
// Swap
void moveZeroes(std::vector<int>& nums)
{
    int zIdx = 0;
    for (auto idx = 0; idx < nums.size(); idx++)
    {
        if (nums[idx] != 0)
        {
            swap(nums[idx], nums[zIdx]);
            
            zIdx++;
        }
    }
}
// 복사만 하고 0을 맨 뒤로 보냄
void moveZeroes2(std::vector<int>& nums)
{
    int zIdx = 0;
    for (auto idx = 0; idx < nums.size(); idx++)
    {
        if (nums[idx] != 0)
        {
            nums[zIdx] = nums[idx];
            zIdx++;
        }
    }
 
    for (; zIdx < nums.size(); zIdx++)
    {
        nums[zIdx] = 0;
    }
}
 
int main()
{
    std::vector<int> nums = { 1,3,0,0,0,4,0,5 };
    moveZeroes(nums);
    for (auto num : nums)
    {
        std::cout << num << " ";
    }
    std::cout << std::endl;
    return 0;
}