[Python] 백준 2153번(소수 단어) 문제 풀이

 

def is_prime(n):
    if n == 1:
        return "It is a prime word."
    for i in range(2, n):
        if n % i == 0:
            return "It is not a prime word."
    return "It is a prime word."

s = 0
for i in input():
    if i.islower():
        s += ord(i) - 96
    else:
        s += ord(i) - 38
print(is_prime(s))

 

 입력받은 문자열에 해당하는 숫자를 s 변수에 더한 후, 그가 소수인지 아닌지를 판별한다. is_prime() 함수는 저번에 쓴 글에서 가져왔고, ord() 함수도 다른 글에 설명해 놓았다. islower() 도 다른 글에 설명해 놓았다. (이제 글이 좀 쌓인 모양이다. 130개 썼는데.)