(cpp) Baekjoon 9184번 문제 ‘신나는 함수 실행’ - 동적계획법

09 Jan 2023 on Programming Languages

Baekjoon 9184번 문제 ‘신나는 함수 실행’ - 동적계획법

---

문제

풀이

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
using namespace std;
int a, b, c;
int result;

//메모이제이션
int arr[21][21][21];
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 w(20, 20, 20);
	}

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

	if (a < b && b < c) {
		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];
	}
	else {
		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);
		return arr[a][b][c];
	}
}
int main() {
	while (true) {
		cin >> a >> b >> c;
		if (a == -1 && b == -1 && c == -1) {
			break;
		}


		printf("w(%d, %d, %d) = %d \n", a, b, c, w(a,b,c));

	}
}