[BOJ] 2941 크로아티아 알파벳

문제

예전에는 운영체제에서 크로아티아 알파벳을 입력할 수가 없었다. 따라서, 다음과 같이 크로아티아 알파벳을 변경해서 입력했다.

예를 들어, ljes=njak은 크로아티아 알파벳 6개(lj, e, š, nj, a, k)로 이루어져 있다. 단어가 주어졌을 때, 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

dž는 무조건 하나의 알파벳으로 쓰이고, d와 ž가 분리된 것으로 보지 않는다. lj와 nj도 마찬가지이다. 위 목록에 없는 알파벳은 한 글자씩 센다.

입력

첫째 줄에 최대 100글자의 단어가 주어진다. 알파벳 소문자와 '-', '='로만 이루어져 있다.

단어는 크로아티아 알파벳으로 이루어져 있다. 문제 설명의 표에 나와있는 알파벳은 변경된 형태로 입력된다.

출력

입력으로 주어진 단어가 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

예제 입력 1

ljes=njak

예제 출력 1

6

예제 입력 2

ddz=z=

예제 출력 2

3

예제 입력 3

nljj

예제 출력 3

3

예제 입력 4

c=c=

예제 출력 4

2

예제 입력 5

dz=ak

예제 출력 5

3

풀이

초기 코드

만약 입력받은 문자의 개수가 1개라면 바로 1을 출력한 뒤 종료한다.

한 문자씩 쪼개어 본다.
마지막 문자일 경우,
개수와 시작점을 1씩 증가시킨다.

맨 뒤에서 두번째 문자일 경우,
현재 위치 ~ 현재 위치 + 1 를 잘라서 그 문자가 두 자리의 크로아티아 문자인지 확인한다.
맞다면 시작점을 2 증가시키고 개수를 1 증가시킨다.
다르다면 시작점을 1 증가시키고 개수를 1 증가시킨다.

나머지 경우,
두문자, 세문자 잘라서 저장해두고 해당 문자가 크로아티아 문자인지 본다.
두자리 크로아티아 문자와 같다면 시작점을 2 증가시키고 개수를 1 증가시킨다.
세자리 크로아티아 문자와 같다면 시작점을 3 증가시키고 개수를 1 증가시킨다.
아니라면 시작점을 1 증가시키고 개수를 1 증가시킨다.

개수를 출력한다.

간단 코드

입력받은 문자열에서 크로아티아 문자를 찾아 해당 문자를 공백으로 바꾼다. (크로아티아 문자의 개수를 1로 줄이는 작업)
다 바꾼 후 변경된 문자열의 개수를 반환한다.

코드

초기 작성 코드

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();
        int count = 0, start = 0;

        // base case
        if (s.length() == 1) {
            System.out.println(1);
            return;
        }

        while (start < s.length()) {
            if (start == s.length() - 1) {
                count++;
                start++;
            } else if (start == s.length() - 2) {
                String curSubStr2 = s.substring(start, start + 2);
                if (curSubStr2.equals("c=") || curSubStr2.equals("c-") || curSubStr2.equals("d-") || curSubStr2.equals("lj") || curSubStr2.equals("nj") || curSubStr2.equals("s=") || curSubStr2.equals("z=")) {
                    start += 2;
                } else {
                    start++;
                }
                count++;
            } else {
                String curSubStr2 = s.substring(start, start + 2);
                String curSubStr3 = s.substring(start, start + 3);
                if (curSubStr2.equals("c=") || curSubStr2.equals("c-") || curSubStr2.equals("d-") || curSubStr2.equals("lj") || curSubStr2.equals("nj") || curSubStr2.equals("s=") || curSubStr2.equals("z=")) {
                    start += 2;
                } else if (curSubStr3.equals("dz=")) {
                    start += 3;
                } else {
                    start++;
                }
                count++;
            }
        }
        System.out.println(count);
    }
}

더 간단한 코드

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();

        // base case
        if (s.length() == 1) {
            System.out.println(1);
            return;
        }

        String[] croatia = {"c=", "c-", "dz=", "d-", "lj", "nj", "s=", "z="};

        for (String i : croatia) {
            s = s.replace(i, " ");
        }

        System.out.println(s.length());
    }
}