12-SeongHoonC

[SeongHoonC]

🔗 문제 링크

분할 정복 한 번 더 정복하기
색종이 만들기

✔️ 소요된 시간

40분

✨ 수도 코드

현재 범위 내 종이가 전부 파란색인지 하얀색인지 검사합니다.

• 둘 중 하나라면 그 색종이 숫자를 +1 하고 함수를 종료합니다.
• 둘 중 어느하나도 아니라면 종이를 1, 2, 3, 4 분면으로 나누어 재귀를 실행합니다.

var blue = 0
var white = 0
fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    val paper = Array(n) { Array(n) { 0 } }

    for (i in 0 until n) {
        val line = br.readLine().split(" ").map { it.toInt() }
        for (j in 0 until n) {
            paper[i][j] = line[j]
        }
    }

    // 재귀 함수 실행
    cut(0, 0, n, paper)
    println(white)
    println(blue)
}

private fun cut(x: Int, y: Int, n: Int, paper: Array<Array<Int>>) {
    // 모두 1이면 파랑 +1 후 return
    //모두 0이면 하양 + 1 후 return
    if (checkPaper(x = x, y = y, n = n, paper = paper, target = 1) { blue++ }) return
    if (checkPaper(x = x, y = y, n = n, paper = paper, target = 0) { white++ }) return

    // n 을 2로 나누어 1, 2, 3, 4 분면 재귀로 실행
    val divN = n / 2
    cut(x, y, divN, paper)
    cut(x + divN, y, divN, paper)
    cut(x, y + divN, divN, paper)
    cut(x + divN, y + divN, divN, paper)
}

private fun checkPaper(x: Int, y: Int, n: Int, paper: Array<Array<Int>>, target: Int, action: () -> Unit): Boolean {
    val isSame = (x until x + n).all { dx -> (y until y + n).all { dy -> paper[dx][dy] == target } }
    if (isSame) {
        action()
        return true
    }
    return false
}

📚 새롭게 알게된 내용

[alstjr7437]

저는 예전에 java로 푼 코드가 있었습니다!!
따로 돌아가는 로직은 kotlin과 똑같은 것 같네요!
다시보니까 저도 n/2를 해놨네요 ㄷㄷ
생각보다 코드가 많이 비슷해서 신기군요!!

import java.util.Scanner;
 
 
public class Main {
	
	public static int white = 0;
	public static int blue = 0;
	public static int[][] board;
 
	public static void main(String[] args) {
		
		Scanner in = new Scanner(System.in);
		
		int N = in.nextInt();
		
		board = new int[N][N];
		
		for(int i = 0; i < N; i++) {
			for(int j = 0; j < N; j++) {
				board[i][j] = in.nextInt();
			}
		}
		
		partition(0, 0, N);
		
		System.out.println(white);
		System.out.println(blue);
		
	}
	
	public static void partition(int row, int col, int size) {
		
		if(colorCheck(row, col, size)) {
			if(board[row][col] == 0) {
				white++;
			}
			else {
				blue++;
			}
			return;
		}
		
		int newSize = size / 2;	
		partition(row, col, newSize);						
		partition(row, col + newSize, newSize);				
		partition(row + newSize, col, newSize);				
		partition(row + newSize, col + newSize, newSize);	
	}
	
	
	public static boolean colorCheck(int row, int col, int size) {
	
		int color = board[row][col];	
		
		for(int i = row; i < row + size; i++) {
			for(int j = col; j < col + size; j++) {
				if(board[i][j] != color) {
					return false;
				}
			}
		}
		return true;
	}
}

그리고 아래는 python으로 다시 푼 코드입니다!

N = int(input())
paper = []
result = [0, 0]
def solution(x,y,n):
    color = paper[x][y]
    for i in range(x, x+n):
        for j in range(y, y+n):
            if color != paper[i][j]:
                solution(x,y,n//2)
                solution(x, y+n//2, n//2)
                solution(x+n//2,y,n//2)
                solution(x+n//2,y+n//2,n//2)
                return 0
    if color == 0:
        result[0] += 1
    else :
        result[1] += 1

for i in range(N):
    paper.append(list(map(int, input().split())))

solution(0,0,N)
print(result[0])
print(result[1])

[alstjr7437]

이렇게 n//2부분을 그냥 보기 편하게 만들 수 있네요!!

[SeongHoonC]

변수를 따로 저장하지 않아도 컴파일러가 최적화해서 여러번 연산이 일어나진 않겠지만..개인적으로 다른 사람이 이해하기 쉽도록 이름 지어주는 편입니당!

[wkdghdwns199]

white_color = 0 
blue_color = 0

def divide(start_x, end_x, start_y, end_y) :
    global white_color, blue_color
    checking_color = paper[start_x][start_y]
    sw=0
    for index_x in range(start_x, end_x):
        for index_y in range(start_y, end_y):
            if paper[index_x][index_y] != checking_color :
                sw=1
                break
        if sw==1 : break
    
    if sw==0 : 
        if checking_color == 1 :
            blue_color = blue_color +1
        else :
            white_color = white_color + 1
    else :
        divide(start_x, int((start_x + end_x)/2), start_y, int((start_y+end_y)/2))
        divide(start_x, int((start_x+end_x)/2), int((start_y+end_y)/2), end_y)
        divide(int((start_x+end_x)/2), end_x, start_y, int((start_y+ end_y)/2))
        divide(int((start_x+end_x)/2), end_x, int((start_y+end_y)/2), end_y)
    return 

        
import sys
input = sys.stdin.readline



N = int(input())
paper = [list(map(int, input().split())) for _ in range(N)]
 
divide(0,N,0,N)


print(white_color)
print(blue_color)