본문 바로가기

PS

개미 전사 [이것이 취업을 위한 코딩테스트다 with 파이썬]

1. 문제에 대한 이해

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

- 문제가 무엇인가?

-> 어떻게 최적의 경우를 선택할 수 있는가?

-> 이것을 DP로 어떻게 푸는가?

-> 100이다

-> 100이면 DFS로 하면 안되지 않는가?

-> .어떻게 케이스를 나누는가?

-> 문제가 무엇인가?

-> 케이스를 잘 나누는가?

-> 잘 나눈다는 것이 무엇인가?

-> 이걸 왜 해야 하는가?

-

    

2. 계획

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

3. 실행

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

[내 풀이] 

 

import java.util.*; 

public class Main {
    
    public static int[] arr;
    public static int[][] dp;
    
    
    public void calculate(int N){
        
          int maxResult = Integer.MIN_VALUE;
          
          dp[1][0] = 0;
          dp[1][1] = arr[1];
          
          dp[2][0] = arr[1];
          dp[2][1] = arr[2];
          
          dp[3][0] = Math.max(dp[1][1], dp[2][1]);
          dp[3][1] = arr[1] + arr[3];
        
        
          for(int i=4; i<=N; i++){
              dp[i][0] = Math.max(dp[i-1][1], dp[i-2][1]);
              dp[i][1] = Math.max(dp[i-1][0]+arr[i], dp[i-2][1]+arr[i]);
              dp[i][1] = Math.max(dp[i][1], dp[i-3][1]+arr[i]);
          }
        
        
          for(int i=4; i<=N; i++){
              maxResult = Math.max(maxResult, dp[i][0]);
              maxResult = Math.max(maxResult, dp[i][1]);
          }
        
        
          System.out.println(maxResult);
        
    }
    
    
    
    
    public static void main(String args[]) {
      Main T = new Main();
      Scanner sc = new Scanner(System.in);
      
      arr = new int[110];
      dp = new int[110][3];
      
      
      int N = sc.nextInt();
      
      for(int i=1; i<=N; i++){
          
          int num = sc.nextInt();
          arr[i] = num; 
      }
      
      T.calculate(N);
      
    }
}

[참고 풀이]

import java.util.*; 

public class Main {
    
    public static int[] arr;
    public static int[] dp;
    
    
    public void calculate(int N){
        
          int maxResult = Integer.MIN_VALUE;
          
          dp[1] = arr[1];
          dp[2] = Math.max(arr[0], arr[1]);
          
          for(int i=3; i<=N; i++){
              dp[i] = Math.max(dp[i-1], dp[i-2]+arr[i]);
          }
          
          System.out.println(dp[N]);
          
    }
    
    
    
    
    public static void main(String args[]) {
      Main T = new Main();
      Scanner sc = new Scanner(System.in);
      
      arr = new int[110];
      dp = new int[110];
      
      
      int N = sc.nextInt();
      
      for(int i=1; i<=N; i++){
          
          int num = sc.nextInt();
          arr[i] = num; 
      }
      
      T.calculate(N);
      
    }
}

4. 반성

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

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

-> i-2번째는 중간에 고려되기 때문에 따로 고려될 필요가 없다

-> 어디서 고려되는가?

-> 선택하지 않는 것이 더 큰 경우에 고려된다

-> 거기에 값이 기록되기 때문이다

-> 따라서 dp를 2칸을 사용할 필요가 없다