[python] 프로그래머스 - 110 옮기기(월간 코드 챌린지 시즌2)

🤔문제 해결

1. 모든 110 찾기

2. 나머지에 0이 있으면 그 0뒤에 110 다 붙이고

3. 0이 없으면 앞에 110을 다 붙인다

 

월코챌 해설지 보고 풀었는데 답 구하는게 헷갈려서 많이 고민한 문제

💻소스 코드

def solution(s):
    answer = []
    for x in s:
        # 모든 110 찾기
        # 110 찾으면 변형된 x에서 또 찾기
        one_one_zero_cnt = 0
        stack = []
        for x_one in x:
            if x_one == '1':
                stack.append(x_one)
            else:
                if len(stack) >= 2 and stack[-1] == '1' and stack[-2] == '1':
                    stack.pop()
                    stack.pop()
                    one_one_zero_cnt += 1
                else:
                    stack.append(x_one)

        x = ''.join(stack)
        # 뒤에서부터 체크해서 0이 나오면 그 0 뒤에 모든 110 붙이기
        # 0 없으면 나머지가 전부 1로 구성되어 있으므로 모든 110을 앞에 붙이기
        for i in range(len(x) - 1, -1, -1):
            if x[i] == '0':
                answer.append(x[:i + 1] + ('110' * one_one_zero_cnt) + x[i + 1:])
                break
        else:
            answer.append('110' * one_one_zero_cnt + x)

    return answer

 

📕문제 확인

출처: 프로그래머스

링크: 110 옮기기