[Programmers] 110 옮기기

https://programmers.co.kr/learn/courses/30/lessons/77886

코딩테스트 연습 - 110 옮기기

 

110은 한 세트로 무조건 움직여야 한다는 개념만 가지면 된다.

그리고 110이 만들어지면 그 이전에건 신경쓰지 않아도 된다.

 

def solution(s):
    answer = []
    for num in s:
        answer.append(go(num))
    return answer
def go(num: str):
    ind = 0
    cnt = 0
    st = []
    numres = ""
    for i in range(len(num)):
        if len(st) > 2 and st[-1] == "0" and st[-2] == "1" and st[-3] == "1":
            for j in range(3):
                st.pop()
            cnt += 1
        st.append(num[i])
    if len(st) > 2 and st[-1] == "0" and st[-2] == "1" and st[-3] == "1":
        for i in range(3):
            st.pop()
        cnt += 1
    for i in range(len(st)):
        numres += st[i]
    res = "110"*cnt
    check = False
    for i in range(len(numres)-1, -1, -1):
        if numres[i] == "0":
            numres = numres[:i+1]+res+numres[i+1:]
            check = True
            break
    if check is False:
        numres = res + numres
    return numres