题目:http://poj.org/problem?id=1579
题目说递归很费时,怎么办?
递归费时的一个很重要的原因就是很多子问题重复计算了,如果计算过的子问题保存下来,用时直接取来用。就是具备记忆能力的递归了,那样就比较快了。
状态转移方程就是题目的递归式了。
代码:
#include <stdio.h> #include <string.h> #define N 21 int ans[N][N][N]; int dp(int a,int b,int c); int main() { int a,b,c; while( ~scanf("%d%d%d",&a,&b,&c) ) { if(-1 ==a && -1 == b && -1 ==c ) break; memset(ans,-1,sizeof(ans)); printf("w(%d, %d, %d) = %d\n",a,b,c,dp(a,b,c)); } return 0; } int dp(int a,int b,int c) { if( a<=0 || b<=0 || c<=0) return 1; if(a>20 || b>20 || c> 20) { if(-1 == ans[20][20][20] ) ans[20][20][20] = dp(20,20,20); return ans[20][20][20]; } if(a<b && b<c) { if(-1 == ans[a][b][c-1]) ans[a][b][c-1] = dp(a,b,c-1); if(-1 == ans[a][b-1][c-1]) ans[a][b-1][c-1] = dp(a,b-1,c-1); if(-1 == ans[a][b-1][c]) ans[a][b-1][c] = dp(a,b-1,c); ans[a][b][c] = ans[a][b][c-1] + ans[a][b-1][c-1]- ans[a][b-1][c]; } else { if(-1 == ans[a-1][b][c]) ans[a-1][b][c] = dp(a-1,b,c); if(-1 == ans[a-1][b-1][c]) ans[a-1][b-1][c] = dp(a-1,b-1,c); if(-1 == ans[a-1][b][c-1]) ans[a-1][b][c-1] = dp(a-1,b,c-1); if(-1 == ans[a-1][b-1][c-1]) ans[a-1][b-1][c-1] = dp(a-1,b-1,c-1); ans[a][b][c] = ans[a-1][b][c]+ans[a-1][b-1][c]+ans[a-1][b][c-1]-ans[a-1][b-1][c-1]; } return ans[a][b][c]; }