[백준 9184: 신나는 함수 실행] - C++

www.acmicpc.net/problem/9184

9184번: 신나는 함수 실행

#include <iostream>
#include <string.h>

using namespace std;

int cache[51][51][51];
int w(int a, int b, int c){


	if (a <= 0 || b <= 0 || c <= 0) {
		return 1;
	}

	int& result = cache[a][b][c];

	if (result != -1) {
		return result;
	}




	if (a >= 21 || b >= 21 || c >= 21)
		return w(20, 20, 20);


	if (a < b && b < c)
	{
		result = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
	}
	else 
	{
		result = 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 result;
}

int main() {

	int a, b, c;
	int n = 0;

	memset(cache, -1, sizeof(cache));
	while (1)
	{

		cin >> a >> b >> c;

		if (a == -1 && b == -1 && c == -1)
			break;

		n = w(a, b, c);

		cout << "w(" << a << ", " << b << ", " << c << ") = " << n << '\n';
	}

	return 0;
}//end of main