[Codeforces] 721A: One-dimensional Japanese Crossword

https://codeforces.com/problemset/problem/721/A

Problem - 721A - Codeforces

코드

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		int n = scan.nextInt();
		String str = scan.next();
		int blackCount = 0;	// 연속된 검정 사각형의 후
		List<Integer> list = new ArrayList<Integer>();
		
		for(int i=0; i<str.length(); i++) {
			// 현재 검정색 블록일 때
			if(str.charAt(i) == 'B') 
				blackCount ++;
			
			// 전에 검정 사각형이 있고, 현재 블록이 화이트일 때
			if(blackCount != 0 && str.charAt(i) == 'W') {
				list.add(blackCount);
				blackCount = 0;
			}
			
			// 마지막 블록까지 도달했을 때, 검정색 블록이 카운트 되어있는경우
			if(i == str.length()-1 && blackCount != 0)
				list.add(blackCount);
		}
		
		System.out.println(list.size());
		for(int i : list)
			System.out.print(i + " ");
		
		scan.close();
	}
}

풀이

1 x n 의 직사각형에서, 연속된 검정색의 갯수와 검정색 그룹의 갯수를 출력하는 문제.

ArrayList를 선언해서 for문을 돌면서 각 검정색의 갯수를 ArrayList에 넣는다.

그룹의 갯수는 list에 넣은 size()와 동일할 것이고, 각각의 요소를 뽑아내면 연속된 검정색의 갯수를 알 수 있다.

현재 검색하고 있는 색이 'W'일 경우, 이전에 검정색으로 칠한게 있는경우(blackCount != 0) list에 넣어준다.

마지막 if문의 경우, 직사각형의 마지막이 W로 끝났을경우, 연속된 검정 블록을 list에 넣지 못하므로 조건을 추가했다.