💡 Codeing Test/백준

백준 5525) IOIOI (python)

밈98 2024. 3. 7. 12:20

https://www.acmicpc.net/problem/5525

 

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

 

import sys
input = sys.stdin.readline

n = int(input())
m = int(input())
S = input().rstrip()

i = 0
cnt = 0
answer = 0
while i < (m - 1):
  if S[i : i+3] == 'IOI':
    i += 2
    cnt += 1
    if cnt == n:
      answer += 1
      cnt -= 1

  else:
    i += 1
    cnt = 0

print(answer)

 

내가 놓친 부분!

 

Pn을 맞춘다음에 찾는 방법을 구현하는 생각이 부족했다

 

Pn의 일부를 포함해 또다른 Pn이 나올 수 있으므로 cnt 0으로 하지 않고 -1로 코드를 구현했어야 했다.