[BOJ / 백준] 14405번 피카츄 python / 파이썬 list 인덱스

출처 : https://www.acmicpc.net/problem/14405

14405번: 피카츄

---

문자열로 입력을 받아서 첫 글자가 p, k, c인지 판단하고

맞을 경우에 피 카 츄 가 맞는지 확인해주는 방식으로 했다.

 

while문의 범위는 tem이라는 변수가 s의 인덱스를 넘어가지 않을 때로 해주었고,

피, 카, 츄를 확인하는 과정에서도 인덱스의 범위 안에서만 진행해주었다.

# BOJ14405 피카츄

S = input()
ans = "YES"
tem = 0
while tem < len(S):
    if S[tem] == "p":
        tem += 1
        if tem >= len(S) or S[tem] != "i":
            ans = "NO"
            break
    elif S[tem] == "k":
        tem += 1
        if tem >= len(S) or S[tem] != "a":
            ans = "NO"
            break
    elif S[tem] == "c":
        tem += 1
        if tem >= len(S) or S[tem] != "h":
            ans = "NO"
            break
        elif tem >= len(S) or S[tem] == "h":
            tem += 1
            if tem >= len(S) or S[tem] != "u":
                ans = "NO"
                break
    else:
        ans = "NO"
        break
    tem += 1
print(ans)