boj16987 계란으로 계란치기

링크

https://www.acmicpc.net/problem/16987

문제

원래 프로그래머의 기본 소양은 팔굽혀펴기를 단 한 개도 할 수 없는 것이라고 하지만 인범이는 3대 500을 넘기는 몇 안되는 프로그래머 중 한 명이다. 인범이는 BOJ에서 틀린 제출을 할 때마다 턱걸이를 5회 하는 기적의 운동 루틴을 통해 뇌와 근육을 동시에 단련한다. 근육을 단련할 때 식단이 정말로 중요하다는 것을 아는 인범이는 탄수화물이 많은 밥이나 빵 따위의 아침 식사를 대신해 단백질이 많은 계란찜을 해먹는다. 계란찜을 먹기 위해서는 계란을 깨야 하는데, 인범이는 힘이 너무 넘치는 나머지 부엌의 대리석을 이용해 계란을 깨면 늘 껍데기가 산산조각나 뒷처리가 너무 어렵게 되곤 한다. 어떻게 하면 계란을 조심스럽게 깰 수 있을까 고민하던 인범이에게 유현이는 굉장히 좋은 해결책을 알려주었다. 바로 계란으로 계란을 치는 것이다. 계란끼리 부딪쳐보니 껍데기가 아주 예쁘게 갈라지는 것을 발견한 인범이는 앞으로 계란으로 계란을 쳐서 식사 준비를 해야겠다고 생각했다. 유현이는 더 나아가 식사 준비를 할 때에도 두뇌를 단련할 수 있는 좋은 퍼즐을 인범이에게 알려주었다.

문제를 소개하기 전, 계란으로 계란을 치게 될 경우 어떤 일이 벌어지는지를 먼저 이해하고 가자. 각 계란에는 내구도와 무게가 정해져있다. 계란으로 계란을 치게 되면 각 계란의 내구도는 상대 계란의 무게만큼 깎이게 된다. 그리고 내구도가 0 이하가 되는 순간 계란은 깨지게 된다. 예를 들어 계란 1의 내구도가 7, 무게가 5이고 계란 2의 내구도가 3, 무게가 4라고 해보자. 계란 1으로 계란 2를 치게 되면 계란 1의 내구도는 4만큼 감소해 3이 되고 계란 2의 내구도는 5만큼 감소해 -2가 된다. 충돌 결과 계란 1은 아직 깨지지 않았고 계란 2는 깨졌다.

유현이가 인범이에게 알려준 퍼즐은 일렬로 놓여있는 계란에 대해 왼쪽부터 차례로 들어서 한 번씩만 다른 계란을 쳐 최대한 많은 계란을 깨는 문제였다. 구체적으로 계란을 치는 과정을 설명하면 아래와 같다.

1. 가장 왼쪽의 계란을 든다.
2. 손에 들고 있는 계란으로 깨지지 않은 다른 계란 중에서 하나를 친다. 단, 손에 든 계란이 깨졌거나 깨지지 않은 다른 계란이 없으면 치지 않고 넘어간다. 이후 손에 든 계란을 원래 자리에 내려놓고 3번 과정을 진행한다.
3. 가장 최근에 든 계란의 한 칸 오른쪽 계란을 손에 들고 2번 과정을 다시 진행한다. 단, 가장 최근에 든 계란이 가장 오른쪽에 위치한 계란일 경우 계란을 치는 과정을 종료한다.

이 과정을 통해 최대한 많은 계란을 깨는 것이 앞으로 인범이가 매일 아침마다 풀게 될 퍼즐이다. 그리고 유현이는 인범이가 찾은 답이 정답이 맞는지 확인해주려고 한다. 일렬로 놓인 계란들의 내구도와 무게가 차례대로 주어졌을 때 최대 몇 개의 계란을 깰 수 있는지 알아맞춰보자.

입력

첫째 줄에 계란의 수를 나타내는 N(1 ≤ N ≤ 8)가 주어진다. 그 다음 N개의 줄에는 계란의 내구도와 무게에 대한 정보가 주어진다. i+1번째 줄에는 왼쪽에서 i번째에 위치한 계란의 내구도 Si
(1 ≤ Si ≤ 300)와 무게 Wi
(1 ≤ Wi ≤ 300)가 한 칸의 빈칸을 사이에 두고 주어진다.

출력

첫째 줄에 인범이가 깰 수 있는 계란의 최대 개수를 출력한다.

나의 풀이

• dfs로 풀이. 예외처리만 잘해주면 되는 문제였고, 예외처리는 아래 코드 주석을 확인해주면 된다.
• 판에있는 계란이 내가 갖고 있는 계란빼고 다 깨져있는것은 깨진 계란의 개수를 매번 세주면서 재귀를 했을때, 계란이 나 빼고 다 깨져있거나 내가 깨져있는 경우만 따로 재귀를 돌리고,
  그 외의 경우는 계란 판을 하나씩 돌면서 깨져있지 않은 계란을 하나씩 쳐주면서 dfs 탐색을 해주면 된다.
• 여기서 cnt 를 broke_cnt로 따로 체크해준 이유는 i 번째 계란을 손에있는 계란으로 부딪힌 경우를 다 순회하고 반환을 해줬을때, i 번째 계란을 부딪히지 않으면서, i+1 번째 계란을 부딪힌 경우를 구해야하기 때문에, cnt 를 ++ 시키면, 재귀가 끝났을때, cnt ++시킨 만큼 다시 — 시켜줘야되는데, (i 번째 계란을 부딪히지 않은 경우를 구해야 하므로) cnt++ 을 몇번 해주었는지 다시 체크하기 번거롭기 떄문에, broken_cnt 로 재귀돌릴때, 더해주었다.
• 핵심 코드

      private static void dfs(int now,int cnt,List<int[]>eggs){
          if(now==N){
              if(cnt>answer) answer = cnt;
              return;
          }
          //판에있는 계란이 내가갖고있는 계란빼고 다 깨져있거나 or 내가들고있는 계란이 깨져있는 경우
          if(cnt>=N-1||eggs.get(now)[0]<=0){
              dfs(now+1,cnt,eggs);
          }else{
              for (int i = 0; i < eggs.size(); i++) {
                  //내가 들고있는 계란이 있는 위치를 가리키는 경우
                  if(i==now) continue;
                  int[] egg_onHand = eggs.get(now);
                  int[] egg = eggs.get(i);
                  //판에 있는 계란이 깨져있는 경우, 다음 계란을 선택함
                  if(egg[0]<=0) continue;
                  
                  int broke_cnt = 0;
                  egg_onHand[0]-= egg[1];
                  egg[0]-= egg_onHand[1];
                  
                  if(egg_onHand[0]<=0) broke_cnt+=1;
                  if(egg[0]<=0)broke_cnt+=1;
                  
                  dfs(now + 1, cnt + broke_cnt,eggs);
                  egg_onHand[0]+=egg[1];
                  egg[0]+=egg_onHand[1];
              }
          }
      }

• 전체 코드

  package Baekjoon.java.silver;
  
  import java.io.BufferedReader;
  import java.io.InputStreamReader;
  import java.util.*;
  
  public class boj16987 {
      static int N,answer;
      public static void main(String[] args) throws Exception{
          BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
  
          N = Integer.parseInt(br.readLine());
  
          boolean[] broken = new boolean[N];
          StringTokenizer st;
          List<int[]> eggs = new ArrayList<>();
  
          for (int i = 0; i < N; i++) {
              st = new StringTokenizer(br.readLine());
              eggs.add(new int[]{Integer.parseInt(st.nextToken()),Integer.parseInt(st.nextToken())});
          }
          dfs(0,0,eggs);
  
          System.out.println(answer);
  
      }
      //egg[0] : 내구성 egg[1] : 무게
      private static void dfs(int now,int cnt,List<int[]>eggs){
          if(now==N){
              if(cnt>answer) answer = cnt;
              return;
          }
          //판에있는 계란이 내가갖고있는 계란빼고 다 깨져있거나 or 내가들고있는 계란이 깨져있는 경우
          if(cnt>=N-1||eggs.get(now)[0]<=0){
              dfs(now+1,cnt,eggs);
          }else{
              for (int i = 0; i < eggs.size(); i++) {
                  //내가 들고있는 계란이 있는 위치를 가리키는 경우
                  if(i==now) continue;
                  int[] egg_onHand = eggs.get(now);
                  int[] egg = eggs.get(i);
                  //판에 있는 계란이 깨져있는 경우, 다음 계란을 선택함
                  if(egg[0]<=0) continue;
                  
                  int broke_cnt = 0;
                  egg_onHand[0]-= egg[1];
                  egg[0]-= egg_onHand[1];
                  
                  if(egg_onHand[0]<=0) broke_cnt+=1;
                  if(egg[0]<=0)broke_cnt+=1;
                  
                  dfs(now + 1, cnt + broke_cnt,eggs);
                  egg_onHand[0]+=egg[1];
                  egg[0]+=egg_onHand[1];
              }
          }
      }
  }

결과