알고리즘/백준

[백준] 1543: 문서 검색 (JAVA)

김호록님 2024. 3. 25. 18:34

문제

세준이는 영어로만 이루어진 어떤 문서를 검색하는 함수를 만들려고 한다. 이 함수는 어떤 단어가 총 몇 번 등장하는지 세려고 한다. 그러나, 세준이의 함수는 중복되어 세는 것은 빼고 세야 한다. 예를 들어, 문서가 abababa이고, 그리고 찾으려는 단어가 ababa라면, 세준이의 이 함수는 이 단어를 0번부터 찾을 수 있고, 2번부터도 찾을 수 있다. 그러나 동시에 셀 수는 없다.

세준이는 문서와 검색하려는 단어가 주어졌을 때, 그 단어가 최대 몇 번 중복되지 않게 등장하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 문서가 주어진다. 문서의 길이는 최대 2500이다. 둘째 줄에 검색하고 싶은 단어가 주어진다. 이 길이는 최대 50이다. 문서와 단어는 알파벳 소문자와 공백으로 이루어져 있다.

출력

첫째 줄에 중복되지 않게 최대 몇 번 등장하는지 출력한다.

 

풀이

처음에는 for문으로 검색 하는 부분을 구현하려고 하다가, 너무 코드가 지저분해지는 것 같아서 찾아봤더니 역시 내장함수가 있었다.

 

s.contains(w) : s 안에 w가 있으면 true, 없으면 false 반환

s.indexOf(w) : s 안에 w가 있으면 시작 인덱스 반환, 없으면 -1 반환

 

생각한 알고리즘은 이렇다.

먼저, s 안에 w가 있을 경우 idx 변수에 시작 인덱스를 저장한다.

s : abababa , w : aba 이면 idx : 2

 

기존 문자열 s에서 w를 없앤다. 시작 인덱스에 w의 길이를 더해서 substring을 사용해야 단어 하나가 온전히 삭제된다.

문자열 하나를 찾았으니 cnt에 1을 더한다.

s : baba , cnt++

 

s 안에 w가 없을 때까지 반복한다.

 

정답 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine(); // 문서
        String w = br.readLine(); // 검색할 단어
        int cnt = 0;

        while (s.contains(w)) { // s안에 w가 있다면
            int idx = s.indexOf(w); // 인덱스를 다음 시작 위치로 변경
            s = s.substring(idx + w.length()); // s를 자른다
            cnt++;
        }
        System.out.println(cnt);
    }
}

 

알게 된 점

- string.contains(word) 

- string.indexOf(word)