인프런 커뮤니티 질문&답변

vkfksaosldk님의 프로필 이미지
vkfksaosldk

작성한 질문수

자바 코딩테스트 - it 대기업 유제

3. "바둑대회" 코드 해설

바둑대회 조합 코드 작성질문드립니다.

작성

·

229

0

조합으로 이렇게 작성했는데, 답이 다르게 나옵니다.
잘못 생각한 부분이 있을까요??

 

import java.util.*;

class Main {
	public static boolean[] visit;
	public static int answer;
	public static int[] white,black;
	public static int sum1,sum2,n;
	public int solution(int[][] cans){
		int answer = Integer.MAX_VALUE;
		
		n=cans.length;
		white = new int[n];
		black = new int[n];
		visit = new boolean[n];
		
		int i=0;
		for(int[] x : cans) { //흰색돌, 검정돌 분리해주기
			white[i] = x[0];
			black[i] = x[1];
			i++;
		}
		
		dfs(0,0); //조합 탐색 시작
		return answer;
	}
	public static void dfs(int L, int s) {
		if(L==n/2) {
			sum1=0;
			sum2=0;
			for(int i=0; i<n; i++) { //방문했던곳은 흰색 돌, 그렇지 않은 곳은 검은돌
				if(visit[i]) sum1+=white[i]; //흰색돌들의 합
				else sum2+=black[i]; //검정색 돌들의 합
			}
			answer = Math.min(answer, Math.abs(sum1-sum2));
		}
		
		else {
			for(int i=s; i<n; i++) {
				if(!visit[i]) {
					visit[i] = true;
					dfs(L+1, i+1);
					visit[i] = false;
				}
			}
		}
	}
	public static void main(String[] args){
		Main T = new Main();
		System.out.println(T.solution(new int[][]{{87, 84}, {66, 78}, {94, 94}, {93, 87}, {72, 92}, {78, 63}}));
		System.out.println(T.solution(new int[][]{{10, 20}, {15, 25}, {35, 23}, {55, 20}}));
		System.out.println(T.solution(new int[][]{{11, 27}, {16, 21}, {35, 21}, {52, 21}, {25, 33},{25, 32}, {37, 59}, {33, 47}}));
	}
}

 

답변 2

0

김태원님의 프로필 이미지
김태원
지식공유자

안녕하세요^^

answer 변수의 선언을 전역으로도 하고 지역으로도 했습니다.

0

안녕하세요, 인프런 AI 인턴이에요.

코드를 보면 answer 변수를 선언한 후 값이 초기화 되지 않은 상태에서 Math.min() 비교가 이루어지기 때문에 최종적인 결과값이 제대로 나오지 않습니다. dfs() 함수에서 모든 경우의 수를 탐색하는 과정 중에 탐색이 끝난 후에 answer 변수에 값을 대입해주어야 합니다.

따라서 dfs() 함수의 if(L==n/2) 블록 안쪽 마지막 부분에 answer = Math.min(answer, Math.abs(sum1-sum2)); 코드를 추가해주어야 합니다.

수정된 코드:

public static void dfs(int L, int s) {
if(L==n/2) {
sum1=0;
sum2=0;
for(int i=0; i; i++) { //방문했던곳은 흰색 돌, 그렇지 않은 곳은 검은돌
if(visit[i]) sum1+=white[i]; //흰색돌들의 합
else sum2+=black[i]; //검정색 돌들의 합
}
answer = Math.min(answer, Math.abs(sum1-sum2));
}

else {
    for(int i=s; i<n; i++) {
        if(!visit[i]) {
            visit[i] = true;
            dfs(L+1, i+1);
            visit[i] = false;
        }
    }
}

}

vkfksaosldk님의 프로필 이미지
vkfksaosldk

작성한 질문수

질문하기