递归的函数
Problem Description
给定一个函数 f(a, b, c):
如果 a ≤ 0 或 b ≤ 0 或 c ≤ 0 返回值为 1;
如果 a > 20 或 b > 20 或 c > 20 返回值为 f(20, 20, 20);
如果 a < b 并且 b < c 返回 f(a, b, c−1) + f(a, b−1, c−1) − f(a, b−1, c);
其它情况返回 f(a−1, b, c) + f(a−1, b−1, c) + f(a−1, b, c−1) − f(a-1, b-1, c-1)。
看起来简单的一个函数?你能做对吗?
Input
输入包含多组测试数据。
对于每组测试数据,输入只有一行为 3 个整数a, b, c(a, b, c < 30)。
My code
#include <stdio.h>
#include <stdlib.h>
int calc(int a, int b, int c);
int func[21][21][21];//store calculated numbers
int main()
{
int a, b, c, f;
func[0][0][0] = 1;
while(~scanf("%d %d %d", &a, &b, &c))
{
if(a <= 0 || b <= 0 || c <= 0)
{
printf("1\n");
}
else if(a > 20 || b > 20 || c > 20)
{
printf("1048576\n");//result of (20,20,20)
}
else
{
func[a][b][c] = calc(a, b, c);
printf("%d\n", func[a][b][c]);
}
}
return 0;
}
int calc(int a, int b, int c)
{
int f;
if(func[a][b][c] == 0)
if(a <= 0 || b <= 0 || c <= 0)
{
return 1;
}
else if(a < b && b < c)
{
f = func[a][b][c] = calc(a, b, c - 1) + calc(a, b - 1, c - 1) - calc(a, b - 1, c);
//printf("%d\n", func[a][b][c]);
}
else
{
f = func[a][b][c] = calc(a - 1, b, c) + calc(a - 1, b - 1, c) + calc(a - 1, b, c - 1) - calc(a - 1, b - 1, c - 1);
//printf("%d\n", func[a][b][c]);
}
else
f = func[a][b][c];
return f;
}
Tips
Save calculated numbers to reduce running time.