기술(Tech, IT)/리트코드(LeetCode)

[LeetCode] 1151. Minimum Swaps to Group All 1's Together

Daniel803 2023. 7. 18. 01:25
class Solution:
    def minSwaps(self, data: List[int]) -> int:
        ret = ones = 0

        for i, d in enumerate(data):
            if d == 1:
                ones += 1

        for i in range(ones):
            if data[i] == 0:
                ret += 1
        
        cnt = ret
        for i in range(ones, len(data)):
            if data[i-ones] == 0:
                cnt -= 1
            if data[i] == 0:
                cnt += 1
            ret = min(ret, cnt)
        
        return ret

 0과 1로 이뤄진 'data'라는 binary array가 input으로 주어질 때 모든 1이 연속하도록 그룹을 지을 때 최소로 swap하는 수를 반환하는 문제다. Two pointer로 시도했으나 잘 되지 않아 Sliding window 방법으로 풀었다. 1의 개수만큼을 Sliding window의 크기로 잡고 매 window에서 0의 개수가 필요한 swap 횟수이므로 0의 개수가 최소가 되는 경우를 찾아 개수를 세 반환하는 방식으로 풀었다.