백준 풀기

백준 2630번: 색종이 만들기 with Java

삼겹살파튀 2024. 8. 18. 23:57

색종이 만들기

 
시간 제한메모리 제한제출정답맞힌 사람정답 비율
1 초 128 MB 49964 34908 26940 69.681%

문제

아래 <그림 1>과 같이 여러개의 정사각형칸들로 이루어진 정사각형 모양의 종이가 주어져 있고, 각 정사각형들은 하얀색으로 칠해져 있거나 파란색으로 칠해져 있다. 주어진 종이를 일정한 규칙에 따라 잘라서 다양한 크기를 가진 정사각형 모양의 하얀색 또는 파란색 색종이를 만들려고 한다.

전체 종이의 크기가 N×N(N=2k, k는 1 이상 7 이하의 자연수) 이라면 종이를 자르는 규칙은 다음과 같다.

전체 종이가 모두 같은 색으로 칠해져 있지 않으면 가로와 세로로 중간 부분을 잘라서 <그림 2>의 I, II, III, IV와 같이 똑같은 크기의 네 개의 N/2 × N/2색종이로 나눈다. 나누어진 종이 I, II, III, IV 각각에 대해서도 앞에서와 마찬가지로 모두 같은 색으로 칠해져 있지 않으면 같은 방법으로 똑같은 크기의 네 개의 색종이로 나눈다. 이와 같은 과정을 잘라진 종이가 모두 하얀색 또는 모두 파란색으로 칠해져 있거나, 하나의 정사각형 칸이 되어 더 이상 자를 수 없을 때까지 반복한다.

위와 같은 규칙에 따라 잘랐을 때 <그림 3>은 <그림 1>의 종이를 처음 나눈 후의 상태를, <그림 4>는 두 번째 나눈 후의 상태를, <그림 5>는 최종적으로 만들어진 다양한 크기의 9장의 하얀색 색종이와 7장의 파란색 색종이를 보여주고 있다.

입력으로 주어진 종이의 한 변의 길이 N과 각 정사각형칸의 색(하얀색 또는 파란색)이 주어질 때 잘라진 하얀색 색종이와 파란색 색종이의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 전체 종이의 한 변의 길이 N이 주어져 있다. N은 2, 4, 8, 16, 32, 64, 128 중 하나이다. 색종이의 각 가로줄의 정사각형칸들의 색이 윗줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 하얀색으로 칠해진 칸은 0, 파란색으로 칠해진 칸은 1로 주어지며, 각 숫자 사이에는 빈칸이 하나씩 있다.

출력

첫째 줄에는 잘라진 햐얀색 색종이의 개수를 출력하고, 둘째 줄에는 파란색 색종이의 개수를 출력한다.

예제 입력 1 복사

8
1 1 0 0 0 0 1 1
1 1 0 0 0 0 1 1
0 0 0 0 1 1 0 0
0 0 0 0 1 1 0 0
1 0 0 0 1 1 1 1
0 1 0 0 1 1 1 1
0 0 1 1 1 1 1 1
0 0 1 1 1 1 1 1

예제 출력 1 복사

9
7

 

 


 

 

첨에 문제를 보고 기절하는 줄 알았다.

근데 알바에서 시루하면서 다시 읽엉보니까 생각보다 공포스러운 문제는..아니더라... 

근까 정사각형 모양으로 다 똑같은 거 아니면 십자가로 썰고 썰고 하라는 거 아닌가요;; 맞제 

내가 생각한 건 첫 항을 보고서 전부 같은 색이면 안 썰기? 색 다르면 1 될 떄까지 존나 썰기? 아 개복잡...

 

컵누들 마라탕맛 개존맛. 마라탕 먹고 싶은데 내일이 걱정돼서 컵누들로 쥐꼬리만한 긴급수혈.

 

어... 한 번 쪼개면 N의 절반 두 번 쪼개면 절반의 절반...... 어...........ㅎ

 

구글링으로 분할정복알고리즘을 찾아냈다^^ 이거 참 저도 아는 알고리즘인데요. 큰 거를 졸라 쪼개서 계산하는 거잖아요. 네 저도 알아요. 그런데 왜 풀지를 못할까요.

 

죄송... 참고를 많이 했어요.... 

쪼개서 합산한다는 게 좀 어려웠음.. (호출을 뭘 어떻게 해야할지) 자료구조 수업 때도 재귀가 쫌 어려웠는데 재귀 공부를 더 해야할 것 같음

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class BJ_2630 {
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
        int N = Integer.parseInt(br.readLine());
        int[][] paper = new int[N][N];
		
        for (int i = 0; i < N; i++) {
            String[] line = br.readLine().split(" ");
            for (int j = 0; j < N; j++) {
                paper[i][j] = Integer.parseInt(line[j]);
            }
        }
        
        int whiteCount = countColor(paper, 0, 0, N, 0);
        int blueCount = countColor(paper, 0, 0, N, 1);
        
        System.out.println(whiteCount); 
        System.out.println(blueCount);  
	}
    
    // 색종이 쪼개고 개수 셈
    static int countColor(int[][] paper, int x, int y, int size, int color) {
        if (isSameColor(paper, x, y, size)) {
            // 전부 같은 색깔이고 color 변수와 같으면 1 return
            return paper[x][y] == color ? 1 : 0;
        } 
        else {
            // 색이 섞여 있으면 4등분으로 분할, 각각의 결과 합산
            int newSize = size / 2;
            return countColor(paper, x, y, newSize, color) // 1사분면
                 + countColor(paper, x, y + newSize, newSize, color) // 2사분면
                 + countColor(paper, x + newSize, y, newSize, color) // 3사분면
                 + countColor(paper, x + newSize, y + newSize, newSize, color); // 4사분면
        }
    }
    
    // 모두 같은 색인지 확인
    static boolean isSameColor(int[][] paper, int x, int y, int size) {
        int color = paper[x][y];
        for (int i = x; i < x + size; i++) {
            for (int j = y; j < y + size; j++) {
                if (paper[i][j] != color) {
                    return false;
                }
            }
        }
        return true;
    }
}