강의

멘토링

커뮤니티

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

우럭아왜우럭님의 프로필 이미지
우럭아왜우럭

작성한 질문수

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

1. 합이 같은 부분집합

합이 같은 부분집합 (total/2)==sum 과 (total-sum)==sum

작성

·

444

0

안녕하세요 강사님.

합이 같은 부분집합 문제에서, 강사님께선 종료조건을
(total - sum) == sum 이걸로 주셨는데,

앞쪽에서 (total / 2) == sum 이것두 된다고하셔서 바꿔서 내봤는데 오답으로 뜨네요,,,

import java.io.*;
import java.util.StringTokenizer;

public class Main {

  static int[] array;
  static int n;
  static int total = 0;
  static boolean isFind = false;
  static String answer = "NO";

  public void DFS(int i, int sum) {
    if (isFind) {
      return;
    }
    if (sum > (total / 2)) {
      return;
    }
    if (i == n) {
      if ((total / 2) == sum) {
        isFind = true;
        answer = "YES";
      }
    } else {
      DFS(i + 1, sum + array[i]);
      DFS(i + 1, sum);
    }
  }

  public static void main(String[] args) throws IOException {
    Main main = new Main();
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    n = Integer.parseInt(br.readLine());
    array = new int[n];
    String s = br.readLine();
    StringTokenizer st = new StringTokenizer(s, " ");
    for (int i = 0; i < n; i++) {
      array[i] = Integer.parseInt(st.nextToken());
      total += array[i];
    }
    main.DFS(0, 0);
    System.out.println(answer);

답변 1

7

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

안녕하세요^^

total/2==sum 이 total값이 홀수임에도 불구하고  '/' 연산자는 몫을 구하기 때문에 참이 될 수 있습니다. 

만약 total이 125이면 125/2==62가 참이 됩니다. 이런 경우를 방지하기 위하여 

if(total%2==0 && (total / 2) == sum) 이렇게 조건을 걸어주면 좋겠습니다.

감사합니다

우럭아왜우럭님의 프로필 이미지
우럭아왜우럭

작성한 질문수

질문하기