백준 알고리즘 20310번 타노스(python)

단순 문자열 문제이다. 1은 앞에서부터, 0은 뒤에서부터 지워주면 된다.

li = list(input())
zero, one = li.count('0')//2, li.count('1')//2
for _ in range(zero):
    li.pop(-li[::-1].index('0')-1)
for _ in range(one):
    li.pop(li.index('1'))
print(''.join(li))