기술(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의 개수가 최소가 되는 경우를 찾아 개수를 세 반환하는 방식으로 풀었다.
반응형