[LeetCode] Move Zeroes

leetcode.com/problems/move-zeroes/

Move Zeroes - LeetCode

5월 이후로 첫 PS 포스팅이다. 사실상 그 동안 한 달에 한 문제 풀까말까한 수준(...)이었다. 코로나19로 인해 의도치 않게 여유가 생긴 김에 조금 풀어 보았다.

문제는 리스트가 주어지고, 원소 중에 0을 모두 맨 뒷부분으로 빼는 문제이다. 단, 새로운 list를 쓰면 안되고 in-place로 풀어야 한다. 새로운 list를 허용하면 for문으로 0이 아닌 원소들을 append하는 단순하고 빠른 풀이가 나오기 때문인 것 같다.

Python을 쓰게 되면서 예전에 풀었던 문제들을 Python으로 좀 풀어보고 있는데, 이 문제는 C++로 이전에 풀었는데도 포스팅을 안해서 같이 하려고 한다.

C++

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int j = 0;
        for(int i = 0;i < nums.size();i++){
            if(nums[i] != 0) continue;
            for(;j < nums.size();j++){
                if(nums[j] == 0) continue;
                if(i > j){
                    j++;
                    i--;
                }else{
                    swap(nums[i], nums[j]);
                }
                break;
            }
        }
    }
};

포인터 2개를 이용해서 0인 원소와 아닌 원소를 교환하는 방식으로 풀었다. list easy단계 대부분의 문제들은 포인터 2개를 이용하면 풀린다.

Python

class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        count = nums.count(0)
        for index in range(count):
            nums.remove(0)
        for index in range(count):
            nums.append(0)

본인이 Modern C++을 잘 모르고 초보라 그런지 몰라도, Python은 기본 list에 대한 함수가 잘 되어있어 0 개수를 세고 그만큼 remove 및 append하는, 위에 비해 굉장히 짧고 단순하게 풀었다.

실행 시간과 메모리는 C++이 더 효율적이었지만, 로직이나 코드의 간단함은 확실히 Python이 좋았다.