작성
·
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
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;
}
}
}
}