[백준 | Java] 14888 연산자 끼워넣기

알린·2024년 4월 2일

baekjoon

목록 보기
44/68

내 풀이

모든 경우를 다 계산 후 최대, 최솟값을 구하는 문제이므로 DFS를 사용했다.
풀이과정은 다음과 같다.

  1. 사칙연산 구호 배열이 0이 아닐 떄 백트래킹과 동시에 사칙연산 결과 넘겨줌
  2. 0이 아닌 사칙연산 구호 배열의 순서에 따라 연산 바뀜
  3. 가장 마지막 정수까지 계산이 완료 되었을 때 최댓값과 최솟값 구해 반환

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    static int N;
    static int[] A;
    static int[] opeCnt = new int[4];
    static int max = Integer.MIN_VALUE;
    static int min = Integer.MAX_VALUE;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        N = Integer.parseInt(br.readLine());
        A = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            A[i] = Integer.parseInt(st.nextToken());
        }

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < 4; i++) {
            opeCnt[i] = Integer.parseInt(st.nextToken());
        }

        dfs(A[0], 1);
        System.out.println(max);
        System.out.println(min);
    }

    static void dfs(int result, int idx) {
        if (idx == N) {
            max = Math.max(max, result);
            min = Math.min(min, result);
            return;
        }
        for (int i = 0; i < 4; i++) {
            if (opeCnt[i] > 0) {
                opeCnt[i]--;

                switch (i) {
                    case 0:
                        dfs(result + A[idx], idx+1);
                        break;
                    case 1:
                        dfs(result - A[idx], idx+1);
                        break;
                    case 2:
                        dfs(result * A[idx], idx+1);
                        break;
                    case 3:
                        dfs(result / A[idx], idx+1);
                        break;
                }
                opeCnt[i]++;
            }
        }
    }
}

profile
짱이 되고싶은 개발 기록

0개의 댓글