[BOJ/백준] 9773 - ID Key [python]

문제 링크

9773번: ID Key

문제 풀이

이 문제는 13자리의 수가 입력되는데, 규칙 두 가지를 통해 나온 값을 더해준 뒤, 4자리로 만들어 출력해주면 된다.

• 각 자릿수를 모두 더해준 값
• 열번째 자리부터 열 세번째 자리까지 세자릿 수에 10을 곱해준 값

두 값의 합을 구해준 뒤, 값을 1000의 자리까지만 출력해준다. 만약, 값이 4자리가 되지 않을 경우, 값에 1000을 더해 출력해준다.

정답 코드

for i in range(int(input())) :
    id = input()
    ans = sum([int(i) for i in id])
    ans += int(id[10 : ]) * 10
    if ans > 9999 : print(str(ans % 10000).zfill(4))
    elif ans < 1000 : print(ans + 1000)
    else : print(ans)