[Python Algorithm] IOIOI BOJ #5525

문제

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

5525번: IOIOI

 

내 문제 풀이

import sys

n = int(sys.stdin.readline())
m = int(sys.stdin.readline())
s = sys.stdin.readline().strip()
idx = 0
count = 0
result = 0

while idx <= m-2:
    if s[idx : idx+3] == "IOI":
        idx += 2
        count += 1
        if count == n:
            result += 1
            count -= 1
    else:
        idx += 1
        count = 0

print(result)

👉 문자열 문제로, 패턴을 찾아서 카운트해주는게 포인트

문제에서 주어지는 P는 'IOI'를 패턴으로 갖는다.
따라서 주어진 s를 돌며 'IOI'패턴이 만족하는 곳을 찾고,

만족한다면 두 칸 뒤로 가서, 또 'IOI'패턴을 만족하는지 검사한다.

검사하면서 count가 n개가 되면 result를 1 증가시키고, count는 1 감소시킨다.

 

💡 피드백

• 처음에는 베스트셀러 문제와 비슷해서 똑같이 풀이했는데,
  이 문제의 경우 IOI라는 특정한 패턴을 가지므로 더 효율적인 풀이가 가능했다.