플밍/문제풀이

백준9184: 신나는 함수 실행(Java)

kkap999 2024. 2. 27. 00:14
728x90

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

 

9184번: 신나는 함수 실행

입력은 세 정수 a, b, c로 이루어져 있으며, 한 줄에 하나씩 주어진다. 입력의 마지막은 -1 -1 -1로 나타내며, 세 정수가 모두 -1인 경우는 입력의 마지막을 제외하면 없다.

www.acmicpc.net

문제 조건 제대로 안보고 풀어서,,, 겁나 틀렸다 ㅠ

오랜만에 재귀 푸니까 재밌당

package BOJ;

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

public class BOJ9184 {
    private static int[][][] arr = new int[21][21][21];

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

        String tmp;
        while (!(tmp = br.readLine()).equals("-1 -1 -1")) {
            StringTokenizer st = new StringTokenizer(tmp);
            int a = parse(st.nextToken());
            int b = parse(st.nextToken());
            int c = parse(st.nextToken());
            System.out.println("w(" + a + ", " + b + ", " + c + ") = " + w(a, b, c));
        }
    }

    private static int w(int a, int b, int c) {
        if (a <= 0 || b <= 0 || c <= 0) {
            return 1;
        }

        if (a > 20 || b > 20 || c > 20) {
            return arr[20][20][20] = w(20, 20, 20);
        }

        if (arr[a][b][c] != 0) {
            return arr[a][b][c];
        }

        if (a < b && b < c)
            return arr[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
        return arr[a][b][c] = w(a - 1, b, c) + w(a - 1, b - 1, c) + w(a - 1, b, c - 1) - w(a - 1, b - 1, c - 1);
    }

    private static int parse(String s) {
        return Integer.parseInt(s);
    }
}

'플밍 > 문제풀이' 카테고리의 다른 글

백준9934 - 완전 이진 트리(Java)  (1) 2024.03.01
백준12919 : A와 B 2  (0) 2024.02.27
백준4179 : 불!  (0) 2023.09.19
백준17106 : 빙고  (0) 2023.06.20
[Java] BOJ1256, 백준 - 사전  (0) 2022.04.11