[백준] 검증수 (Python)

문제

컴퓨터를 제조하는 회사인 KOI 전자에서는 제조하는 컴퓨터마다 6자리의 고유번호를 매긴다. 고유번호의 처음 5자리에는 00000부터 99999까지의 수 중 하나가 주어지며 6번째 자리에는 검증수가 들어간다. 검증수는 고유번호의 처음 5자리에 들어가는 5개의 숫자를 각각 제곱한 수의 합을 10으로 나눈 나머지이다.

예를 들어 고유번호의 처음 5자리의 숫자들이 04256이면, 각 숫자를 제곱한 수들의 합 0+16+4+25+36 = 81 을 10으로 나눈 나머지인 1이 검증수이다.

입력

첫째 줄에 고유번호의 처음 5자리의 숫자들이 빈칸을 사이에 두고 하나씩 주어진다.

출력

첫째 줄에 검증수를 출력한다.

예제 입력 1

0 4 2 5 6

예제 출력 1

1

초기 코드

a, b, c, d, e = map(int, input().split())
    
result = (a ** a + b ** b + c ** c + d ** d + e  ** e)%10
    
print(result)

어쨋든 입력할 숫자가 또 공백을 기준으로 5개의 숫자를 입력하므로 input()의 값을 split() 공백 기준으로 나눠준 후 int화 시켜서 정수로 만든 값으로 만들어서 a,b,c,d,e에 넣어준다
그 다음에 제곱의 합인데 a ** a 이런식으로 기재하여 원하는 로직이 아닌 값을 넣어주게 되었었다

수정 코드

a, b, c, d, e = map(int, input().split())
    
result = (a ** 2 + b ** 2 + c  ** 2 + d ** 2 + e ** 2)%10
    
print(result)