합이 같은 부분 집합(DFS: 아마존 인터뷰) [자바 알고리즘 문제 풀이]

1. 문제에 대한 이해

• 우리가 풀어야 할 문제는 무엇인가?
• 주어진 자료는 무엇인가?
• 조건은 무엇인가?
• 우리가 문제를 풀기 위해 주어진 자료가 충분한가?
• 숨겨진 조건이나 자료가 있는가? 그렇다면 그 것을 다른 방법으로 해석해보라. 

    - 우리가 풀어야 할 문제는 무엇인가?

-> N개의 원소로 구성도니 자연수 집합이 주어지면, 이 집합을 두 개의 부분집합으로 나누었을 때,

    두 부분집합의 원소의 합이 서로 같은 경우가 존재하면  YES를 출력하고, 

    그렇지 않으면 NO를 출력하는 프로그램을 작성하라 

 

2. 계획

• 전에 비슷한 문제를 알고 있는가?
• 이 문제를 푸는데 있어서 유용하게 쓸 수 있는 지식은 무엇인가?
• 비슷한 문제를 풀어본 적이 있다면 그 것을 활용할 수 있는가?
• 만약 문제를 풀 수 없다면 문제를 더 단순하게 하기 위해서 주어진 조건을 버려보아라
• 주어진 자료로부터 유용한 것을 이끌어 낼 수 있는가?
• 자료는 모두 사용했는가?
• 조건을 모두 사용했는가?
• 문제에 포함된 핵심적인 개념은 모두 고려했는가?

3. 실행

• 풀이 계획을 실행하고, 각 단계가 올바른지 점검하라.

import java.util.*;

public class Main {
    
    public static int[] arr;
    public static int N;
    public static int answer; 
    public static boolean[] selected;
    public static int total; 
    
    
    public void DFS(int pos){
        
        if(pos == N){
            
            int sum = 0;
            
            for(int i=0; i<N; i++){
                if(selected[i] == true){
                    sum += arr[i];
                }
            }
            if(sum*2 == total){
                answer++; 
            }
            
            return; 
        }
        
        
        selected[pos] = true;
        DFS(pos+1);
        selected[pos] = false;
        DFS(pos+1);
        
        
    }
    
    
   
    
    public static void main(String args[]) {
      Main T = new Main();
      Scanner sc = new Scanner(System.in);
      
      N = sc.nextInt();
      selected = new boolean[N+1];
      answer = 0; 
      arr = new int[N+1];
      
      for(int i=0; i<N; i++){
          int num = sc.nextInt();
          arr[i] = num; 
          total += num; 
      }
      
      T.DFS(0);
      
      if(answer >= 1){
          System.out.println("YES");
      }else{
          System.out.println("NO");
      }
      
    }
}

[강의 코드]

import java.util.*;

public class Main {
    
    public static int[] arr;
    public static int N;
    public static int answer; 
    public static boolean[] selected;
    public static int total; 
    public static boolean flag;
    
    public void DFS(int pos){
        
        if(flag){
            return; 
        }
        
        if(pos == N){
            
            int sum = 0;
            
            for(int i=0; i<N; i++){
                if(selected[i] == true){
                    sum += arr[i];
                }
            }
            if(sum*2 == total){
                System.out.println("YES");
                flag = true;
            }
            
            return; 
        }
        
        
        selected[pos] = true;
        DFS(pos+1);
        selected[pos] = false;
        DFS(pos+1);
        
        
    }
    
    
   
    
    public static void main(String args[]) {
      Main T = new Main();
      Scanner sc = new Scanner(System.in);
      
      N = sc.nextInt();
      selected = new boolean[N+1];
      answer = 0; 
      arr = new int[N+1];
      flag = false; 
      
      for(int i=0; i<N; i++){
          int num = sc.nextInt();
          arr[i] = num; 
          total += num; 
      }
      
      T.DFS(0);
      
      if(!flag){
          System.out.println("NO");
      }
      
    }
}

4. 반성

• 문제를 다른 방식으로 해결할 수 있는가?
• 결과나 방법을 어떤 다른 문제에 활용할 수 있는가?
• 어떻게 하면 더 효율적으로 문제를 해결할 수 있는가?
• 어떻게 하면 더 효과적으로 문제를 해결할 수 있는가?

- 어떻게 하면 더 효율적으로 문제를 해결할 수 있는가?

-> flag를 둬서 true로 변환하면 하나라도 발견하면 다 돌 필요 없이 그냥 끝낼 수 있어서

    더 효율적이다.