-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathBOJ_5525_IOIOI.java
62 lines (52 loc) · 1.27 KB
/
BOJ_5525_IOIOI.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import java.util.ArrayList;
import java.util.Scanner;
public class BOJ_5525_IOIOI {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int N = sc.nextInt(); // N
int M = sc.nextInt(); // SÀÇ ±æÀÌ
String S = sc.next();
StringBuilder sb = new StringBuilder("I");
for (int i = 0; i < N; i++) {
sb.append("OI");
}
System.out.println(kmp(S, sb.toString()).size());
}
static ArrayList<Integer> kmp(String str, String pattern) {
ArrayList<Integer> list = new ArrayList<>();
int[] pi = getPi(pattern);
int n = str.length(), m = pattern.length(), j = 0;
char[] s = str.toCharArray();
char[] p = pattern.toCharArray();
for (int i = 0; i < n; i++) {
while (j > 0 && s[i] != p[j]) {
j = pi[j - 1];
}
if (s[i] == p[j]) {
if (j == m - 1) {
list.add(i - m + 1);
j = pi[j];
} else {
j++;
}
}
}
return list;
}
static int[] getPi(String pattern) {
int len = pattern.length();
int j = 0;
char[] p = new char[len];
int[] pi = new int[len];
p = pattern.toCharArray();
for (int i = 1; i < len; i++) {
while (j > 0 && p[i] != p[j]) {
j = pi[j - 1];
}
if (p[i] == p[j]) {
pi[i] = ++j;
}
}
return pi;
}
}