알고리즘 공부[Javascript]/백준

[백준] 5525번 / 실버2 / IOIOI / Node.js

Kevinkb 2021. 9. 24. 01:15

문제

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가 주어진다.
1 ≤ N ≤ 1,000,000
2N+1 ≤ M ≤ 1,000,000
S는 I와 O로만 이루어져 있다.

출력

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

예제 입력

1
13
OOIOIOIOIIOII

예제 출력

4

소스 코드

const input = require('fs').readFileSync('/dev/stdin').toString().trim().split('\n');
const N = +input[0];
const M = +input[1];
const S = input[2];
let count = 0;

function solution(input) {
    for (let i = 0; i < M - 2; i++) {
        if (S[i] === 'I') {
            let number = 0;

            while(S[i+1] === 'O' && S[i+2] === 'I') {
                i += 2;
                number++;
                if (number === N) {
                    count++;
                    number--;
                }
            }
        }
    }
    console.log(count);
}

solution(S);