[ BOJ / Python ] 14405번 피카츄

황승환·2021년 12월 28일
0

Python

목록 보기
65/498

이번 문제는 입력받은 문자열에서 pi, ka, chu가 존재하면 문자열에서 해당 부분을 없애는 방식으로 해결하였다.

  • 문자열 s를 입력받는다.
  • pi, ka, chu를 담은 배열 pikachu를 선언한다.
  • while문에 사용할 i를 0으로 선언한다.
  • i가 s의 길이보다 작거나 같을 동안 반복하는 while문을 돌린다.
    -> 만약 s[:i]가 pikachu안에 있을 경우,
    --> s를 s[i:]로 갱신한다.
    --> i를 0으로 초기화한다.
    -> i를 1 증가시킨다.
  • 만약 s의 길이가 0이라면 YES를 출력한다.
  • s의 길이가 0이 아니라면 NO를 출력한다.

Code

s=str(input())
pikachu=['pi', 'ka', 'chu']
i=0
while i <= len(s):
    if s[:i] in pikachu:
        s=s[i:]
        i=0
    i+=1
if len(s)==0:
    print('YES')
else:
    print('NO')

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글