[BaekJoon] 15873번 : 공백 없는 A+B (Python)

코딩 1일 1문제! 오늘의 문제는 공백 없는 A+B입니다.

15873번: 공백 없는 A+B

👨🏻‍💻 코드 ( Solution )

def no_space_A_plus_B(AB):
    answer = 0
    if len(AB) == 2:
        answer = int(AB[0]) + int(AB[1])
    elif len(AB) == 3:
        A = int(AB[0]) if AB[1] != '0' else int(AB[:2])
        B = int(AB[2]) if A == 10 else int(AB[1:])
        answer = A + B
    elif len(AB) == 4:
        answer = 20
        
    return answer
        
if __name__ == "__main__":
    AB = input()
    print(no_space_A_plus_B(AB))

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07