[백준 5525] IOIOI (실버1) Python 풀이

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이 몇 군데 포함되어 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. 둘째 줄에는 S의 길이 M이 주어지며, 셋째 줄에 S가 주어진다.

출력

S에 PN이 몇 군데 포함되어 있는지 출력한다.

---

풀이

포인터를 2칸씩 이동해가며 'IOI' 패턴을 찾는다. 그리고 IOI 패턴의 수만큼 카운트 값을 누적해가며 정답을 리턴하면 된다.

 

# 백준 5525 IOIOI 실버1 https://www.acmicpc.net/problem/5525
# I와 O가 교대로 나오는 문자열 P_N이 S 안에 몇 군데 포함되어 있는지 구하라.
import re
n = int(input())
m = int(input())
s = input()

answer = 0
i = 0
cnt = 0

while i < (m-1):
    if s[i:i+3] == 'IOI': # 3칸씩 탐색하며 IOI 패턴을 찾는다.
        cnt += 1 # IOI 패턴을 찾았다면 cnt에 추가해준다.
        i += 2 # 다음에 탐색할 인덱스로 이동한다.
        if cnt == n: # cnt 값이 n에 도달하면 
            answer += 1 # 정답에 1을 추가해준다.
            cnt -= 1 # 다음에 탐색할 인덱스로 이동할 것이니 이전 IOI 패턴은 카운트에서 제외해준다.
    else:
        cnt = 0 # IOI 패턴이 끊기거나, 시작되지 않았으므로 cnt는 0으로 초기화해준다.
        i += 1 # IOI 패턴을 발견하지 않았다면 1칸만 옮긴다.
        
print(answer)