[Codetree] 이상한 진수 2 Python

https://www.codetree.ai/missions/5/problems/awkward-digits-2/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

import sys


input = sys.stdin.readline

def natural(num):
    result = 0
    e = 1
    for n in num[::-1]:
        result += int(n)*e
        e *= 2
    return result


max_val = 0

binary = list(input().rstrip())

origin = binary[::]

for i in range(len(binary)):
    temp = binary[i]

    binary[i] = '0'
    if origin != binary:
        max_val = max(max_val, natural(binary))
    
    binary[i] = '1'
    if origin != binary:
        max_val = max(max_val, natural(binary))

    binary[i] = temp

        
print(max_val)