[백준 20310 🥈] 타노스

https://www.acmicpc.net/problem/20310

20310번: 타노스

문제 풀이

import sys

numbers=list(sys.stdin.readline().strip())
zero, one = numbers.count('0')//2, numbers.count('1')//2

# 0은 뒤에서부터 지워짐 표시
for idx in range(len(numbers)-1,-1,-1):
    if zero<=0:
        break
    if numbers[idx]=='0':
        numbers[idx]=-1
        zero-=1
# 1은 앞에서부터 지워짐 표시
for idx in range(len(numbers)):
    if one<=0:
        break
    if numbers[idx]=='1':
        numbers[idx]=-1
        one-=1

for idx in range(len(numbers)):
    if numbers[idx]!=-1:
        print(numbers[idx], end='')

• 간단한 문자열 문제였지만, 기존 배열의 배치를 바꿀 수 없다는 조건을 간과해서 처음엔 25점을 맡았다!
• 처음엔 0과 1의 개수를 count해서 2로 나눠서 저장하고, 사전순으로 가장 빠른 경우를 출력해야 하기 때문에 초기에 '0'*(0의 개수)+'1'*(1의 개수)를 출력했었다. (문제를 똑바로 읽고 이해하자ㅠㅠ)
• 기존 배열의 배치를 바꾸지 않으면서 사전순으로 가장 빠른 문자열을 출력하기 위해 '0'은 지워야 하는 0의 개수만큼 뒤에서부터 지워주고, '1'은 1의 개수만큼 앞에서부터 지워주면 0이 앞쪽에 최대한 위치할 수 있고, 1이 최대한 뒤쪽에 위치할 수 있기 때문에 사전순으로 가장 빠른 문자열을 구할 수 있다. 
• 여기서 문자열을 지운다고 단순히 pop(지워야 하는 인덱스)로 처리해주면 pop해서 지워지는 배열에 따라 이미 탐색한 인덱스를 또 지우거나, 지우면 안되는 인덱스를 지울 수 있기 때문에 numbers 배열에 삭제되었다는 처리를 -1로 표시했다.
• 그리고 모든 연산을 끝낸 후에 numbers 값이 -1이 아닌 요소들을 출력해주면 된다.