코딩테스트 대비/BOJ

[Baekjoon/Python] 5525번: IOIOI - 효과는 굉장했다!

bluetag_boy 2021. 11. 23. 03:01
반응형
 

5525번: IOIOI

N+1개의 I와 N개의 O로 이루어져 있으면, I와 O이 교대로 나오는 문자열을 PN이라고 한다. P1 IOI P2 IOIOI P3 IOIOIOI PN IOIOI...OI (O가 N개) I와 O로만 이루어진 문자열 S와 정수 N이 주어졌을 때, S안에 PN이 몇

www.acmicpc.net

 

알고리즘 분류

  • 문자열

 

 

SOLUTION

import sys

N = int(sys.stdin.readline())
M = int(sys.stdin.readline())
S = list(sys.stdin.readline())

answer = 0
cnt = 0
i = 1

while i < M - 1:
    if S[i-1] == "I" and S[i] == "O" and S[i+1] == "I": # IOI가 탐색될때마다 cnt += 1
        cnt += 1

        if cnt == N: # N == 1 -> IOI, N == 2 -> IOIOI, N == 3 -> IOIOIOI ...
            cnt -= 1
            answer += 1 

        i += 1

    else:
        cnt = 0

    i += 1

print(answer)