题目链接:http://poj.org/problem?id=1579
Hint:这是一道简单的DP题。注意全局变量的数组要开得大一些,自己刚开始开得是f[20][20][20],结果下面写的循
环却是到f[21][21][21].另外还要严格注意输出格式。
这本身就是一个递归函数,要是按照函数本身写递归式,结果肯定是TLE,这里我开了一个三维数组,从
f(0,0,0)开始递推,逐步产生到w(20,20,20)的值,复杂度O(n^3).
总结:这道题是很地道的DP,因为它的子问题实在是太多了,所以将问题的结果保存起来,刘汝的书中讲到自底
向上的递推,这个例子就非常典型。本题的思想就是地道的动态规划。
代码如下:
#include<iostream>
#include<fstream>
#include<vector>
#include<string>
#include<map>
#include<iterator>
#include<algorithm>
#include<numeric>
#include<cmath>
#include<sstream>
using namespace std;
long long f[21][21][21];
int main()
{
#ifdef ONLINE_JUDGE
#else
freopen("D:\\in.txt", "r", stdin);
freopen("D:\\out.txt", "w", stdout);
#endif // ONLINE_JUDEG
for (int a = 0; a < 21; a++)
for (int b = 0; b < 21;b++)
for (int c= 0; c < 21; c++)
{
if (a == 0 || b == 0 || c == 0)
{
f[a][b][c] = 1;
}
else if (a<b && b<c)
{
f[a][b][c] = f[a][b][c - 1] + f[a][b - 1][c - 1] - f[a][b - 1][c];
}
else
{
f[a][b][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];
}
}
int a, b, c;
while (scanf("%d%d%d", &a, &b, &c) != EOF)
{
if (-1 == a&&-1 == b&&-1 == c)
{
break;
}
else if (a <= 0 || b <= 0 || c <= 0)
{
printf("w(%d, %d, %d) = 1\n",a,b,c);
}
else if (a > 20 || b > 20 || c > 20)
{
printf("w(%d, %d, %d) = %d\n", a, b, c,f[20][20][20]);
}
else
{
printf("w(%d, %d, %d) = %d\n",a,b,c,f[a][b][c]);
}
}
return 0;
}