19-hadongun

[hadongun]

🔗 문제 링크

IOIOI

✔️ 소요된 시간

47min

✨ 수도 코드

지나가다 보인 문자열 문제를 가져왔습니다.

문제에서 나온 P(N) 수열은 I가 홀수 인덱스에, O는 짝수 인덱스에 나오게 됩니다. 이 부분을 기억해주십시오

주어진 수열에서 P(N) 수열이 몇 번 등장하는가! 를 구해야해요

주어진 수열 s를 모두 확인하기 위해 반복문 while에 조건( i < 전체 길이 - p(n)길이 - 1 )을 설정합니당
위 조건보다 i가 커지면 더 이상 P(N) 수열이 등장 할 수 없기 때문입니다.

아무튼! 위에서 말했던 I, O가 나오는 인덱스를 기억하여
I는 [i + 2j + 1], O는 [i + 2j + 2] 위치에 나오게 됩니다.

P(N) 수열에는 I가 N+1, O가 N개가 총 등장합니다.
처음에 I가 등장하면 이후 반복문으로 진입하도록 설계합니다. -> 남은 I, O 등장 개수는 각각 N, N
아래 반복문이 총 N번 반복하도록 하여 I는 [i + 2j + 1], O는 [i + 2j + 2] 이 s수열 인덱스에 맞게 등장하는지 확인.
등장하지 않으면 is_pass = False로 바꾸어 바로 반복을 멈추고 카운트 하지 않게 합니다.

이렇게 하여 모든 경우를 세어 출력합니당

水導code

입력 N 받기
입력 s_len 받기
입력 문자열 s를 리스트로 저장하기

i를 0으로 초기화하기
count를 0으로 초기화하기

i가 s_len - 2*N 보다 작을 동안 반복하기:
    만약 s[i]가 'I'이면:
        is_pass를 True로 초기화하기
        j를 0부터 N-1까지 반복하기:
            만약 s[i + 2*j + 1]이 'O'가 아니면:
                is_pass를 False로 바꾸고 반복문 나가기
            만약 s[i + 2*j + 2]가 'I'가 아니면:
                is_pass를 False로 바꾸고 반복문 나가기
        반복이 끝난 후:
            만약 is_pass가 True이면:
                count를 1 증가시키기
                i를 2 증가시키기
            아니면:
                i를 1 증가시키기
    아니면:
        i를 1 증가시키기

count 출력하기

📚 새롭게 알게된 내용

[dohyeondol1]

오호 생각못한 흥미로운 풀이네요!
하지만 N이나 M이 커지면, 서브태스크 2에서는 시간초과가 날 것 같아요.

저는 문자열의 각 문자에 대해 반복문을 돌려 현재 문자가 I인 경우 뒤에 OI가 몇개씩 오는지 카운트 했습니다.
P(N)은 O가 N개 라고도 할 수 있지만, 첫글자 I를 제외하면 OI가 N개 라고도 할 수 있씁니다!!!!

OI의 개수가 N과 일치하면 다음 문자에 또 OI가 있을 경우를 생각해
정답에 경우의수 1을 더하고 현재 OI카운트를 1빼줍니다.

예를 들어, P(4) = IOIOIOIOI 인 문자열에서, P(3)을 구해야 할 때,
IOIOIOI + OI 와 IO + IOIOIOI로 바라볼 수 있습니다.

첫번째 경우에서 정답을 발견했다고 현재 카운트를 쉽사리 초기화하면 안되는 것이죠...

#include <iostream>
#include <string>
using namespace std;

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    
	int N, M;
    string S;
	cin >> N >> M >> S;
	
	int answer = 0, OIcount = 0;
	for(int i = 0; i < M; i++) {
	    if(S[i] == 'O') continue;
	    else {
	        while(S[i+1] == 'O' && S[i+2] == 'I') {
	            OIcount++;
	            if(OIcount == N) {
	                answer++;
	                OIcount--;
	            }
	            i += 2;
	        }
	    }
	    OIcount = 0;
	}
	
	cout << answer << '\n';
	return 0;
}

동윤님도 100점을 향해 도전~!!!!!

[froglike6]

앞서 잘 말씀해주셨다시피 동윤님의 코드는 $\mathcal{O}(M\times N)$이라 시간초과가 날 수 있습니다. 그래서 저는 while을 통해 3개의 문자만 보고 판단하도록 문제를 풀어봤습니다.

import sys
input = sys.stdin.readline

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

cnt, ans, i = 0, 0, 0

while i<m-2:
    if s[i]=="I" and s[i+1]=="O" and s[i+2]=="I":
        cnt+=1
        if cnt >= n:
            ans+=1
        i+=2
    else:
        cnt=0
        i+=1
print(ans)

[Fnhid]

저는 kmp로 풀어보았습니다!
뭔가 될 것 같아서 해봤는데 가능하군요 역시 kmp

#include <iostream>
#include <string>
#include <vector>

using namespace std;

vector<int> getPi(string &pattern) {
  int m = pattern.size();
  vector<int> pi(m, 0);
  int j = 0;

  for (int i=1;i<m;++i) {
    while (j > 0 && pattern[j] != pattern[i]) j = pi[j-1];
    if (pattern[j] == pattern[i]) pi[i] = ++j;
  }
  return pi;
}

int kmp(string &pattern, string &text) {
  int cnt=0;
  vector<int> pi = getPi(pattern);
  int n=text.size(), m=pattern.size(), j=0;
  for (int i=0;i<n;i++) {
    while (j>0 && pattern[j] != text[i]) j = pi[j-1];
    if (pattern[j] == text[i]) {  // pattern matching success
      if (j == m - 1) {
        cnt++;
        j = pi[j];
      } else {
        j++;
      }
    }

  }
  return cnt;
}

int main() {
  cin.tie(NULL);
  ios::sync_with_stdio(false);
  int n, m, nCount, j;
  string s, patt="I";
  cin >> n >> m >> s;

  for (int i =0;i<n;i++) {
    patt+="OI";
  }
  cout << kmp(patt, s);
  return 0;
}