[백준] 9324번: 진짜 메시지(구현)

https://www.acmicpc.net/problem/9324

9324번: 진짜 메시지

코드

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		for(int tc=0; tc<t; tc++) {
			String str = scan.next();
			int[] alpa = new int[26];
			boolean flag = true;
			String result = "";
			for(int i=0; i<str.length(); i++) {
				alpa[str.charAt(i) - 'A'] ++;
				
				// 동일한 알파벳이 3번 등장했을 경우,
				if(alpa[str.charAt(i) - 'A'] == 3) {
					
					// 마지막 문자에 도달했을때 동일한 문자가 3번이면 문자가 삽입이 안된경우이므로 틀린 문자.
					if(i == str.length()-1) {
						flag = false;
						break;
					}
					
					// 문자를 한 번 삽입해야하는데 앞뒤 다르면 틀린 문자.
					if(str.charAt(i) != str.charAt(i+1)) {
						flag = false;
						break;
					}
					
					// 위 케이스에 모두 포함안되면 세 번 등장한 문자가 삽입된 경우므로, 한칸 건너뛴다.
					i ++;
					
					// 해당 문자 횟수 초기화 시켜주기
					alpa[str.charAt(i) - 'A'] = 0;
				}
			}
			result = (flag) ? "OK" : "FAKE";
			System.out.println(result);
			
		}
		
		scan.close();
	}

}

풀이

각 문자가 세 번 등장할 때 한 번 더 문자가 삽입된다.

주어진 문자열이 끝날때까지 위 조건을 만족하면 진짜 메시지고, 만족하지 못하면 가짜 메시지다.

"AABCAABBB"

위에서 색칠한 A, B같은경우 이전에 문자가 3번왔으므로 한 번 더 삽입된 경우다.

 

따라서 문자열을 끝까지 탐색하면서 아스키코드값을 기준으로 해당 문자를 인덱스로 배열에 값을 증가시킨다.

alpa[str.charAt(i) - 'A'] ++;

 

 

1) 그 후 문자열이 3번왔을경우, 예제 입력 2번째와 같이 문자열이 마지막인 경우도 고려해야한다.

// 마지막 문자에 도달했을때 동일한 문자가 3번이면 문자가 삽입이 안된경우이므로 틀린 문자.
if(i == str.length()-1) {
    flag = false;
    break;
}

 

 

2) 만약 3번 온 문자와 다음 문자가 다를경우

AAA -> AAAB

위와 같은경우는 조건을 만족하지 못하므로 가짜 메시지다.

// 문자를 한 번 삽입해야하는데 앞뒤 다르면 틀린 문자.
if(str.charAt(i) != str.charAt(i+1)) {
    flag = false;
    break;
}

 

 

3) 위 케이스에 포함이 안될경우, 문자가 하나더 삽입된 경우이므로 한칸 뛰어넘고, 해당 문자의 횟수를 초기화한다.

// 위 케이스에 모두 포함안되면 세 번 등장한 문자가 삽입된 경우므로, 한칸 건너뛴다.
i ++;
					
// 해당 문자 횟수 초기화 시켜주기
alpa[str.charAt(i) - 'A'] = 0;

 

 

4) 마지막으로 flag 값을 기준으로 true면 진짜 메시지, false면 가짜 메시지이다.

result = (flag) ? "OK" : "FAKE";
System.out.println(result);