[BOJ/백준] 6750 - Rotating letters [python]

문제 링크

6750번: Rotating letters

문제 풀이

입력되는 문자열 중 I, O, S, H, Z, X, N이 아닌 문자가 포함되어있는 경우, NO를 출력하고 아니라면 YES를 출력한다.

정답 코드

for i in input() :
    if i not in ['I', 'O', 'S', 'H', 'Z', 'X', 'N'] :
        print("NO")
        exit(0)
print("YES")