[파이썬] 백준 알고리즘 No.5525 IOIOI

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

5525번: IOIOI

 

최초 코드

import sys
input = sys.stdin.readline

n = int(input())
m = int(input())
s = input()

p = 'IO' * n + 'I'
p_length = len(p)
count = 0

for i in range(m - p_length + 1):
    if s[i] == 'I':
        if s[i: i + p_length] == p:
            count += 1
print(count)

시간 초과..

 

 

 

조금더 빨리 바꿔보자

import sys
input = sys.stdin.readline

n = int(input())
m = int(input())
s = input()

answer, i, count = 0, 0, 0

while i < (m - 1):
    if s[i : i + 3] == 'IOI':
        i += 2
        count += 1
        if count == n:
            answer += 1
            count -= 1
    else:
        i += 1
        count = 0
print(answer)