前言
本题依旧是个很简单的基础题(可能是普及题做多了)
先给出题目位置:P1464 Function
解题思路:
本题是一道简单的递归题,而且递归求解方法题目也给了,但是由于存在大量的重叠子问题,因此,当输入的a,b,c稍微大一点,时间开销就会很大。因此,本题主要还是想推荐一个新的思路:备忘录。分析原题目,发现有效的w(a,b,c)只在
0
≤
a
,
b
,
c
≤
20
0\le a,b,c \le20
0≤a,b,c≤20范围内有效,所以用一个
w
[
21
]
[
21
]
[
21
]
w[21][21][21]
w[21][21][21]的三维数组就可存下来(本质上C语言里的n维数组只是一维数组,这是C语言的特性,在这里就不多赘述了)。然后递归的过程中,先在
w
[
21
]
[
21
]
[
21
]
w[21][21][21]
w[21][21][21]里查找相应的
w
w
w向量是否存在,若不存在,则递归向下查找,否则就直接从
w
[
21
]
[
21
]
[
21
]
w[21][21][21]
w[21][21][21]对应位置处取相应值即可。
w
[
]
[
]
[
]
w[][ ][ ]
w[][][]即为备忘录。这样可以有效解决重叠子问题导致的时间复杂度开销。下面给出我的AC代码:
#include <stdio.h>
#include <stdlib.h>
#define inf 140000000
long long int w[21][21][21];
long long int w_num(long long int a,long long int b,long long int c){
if(a<=0||b<=0||c<=0){
return 1;
}else if(a>20||b>20||c>20){
if(w[20][20][20]==inf){
w[20][20][20]=w_num(20,20,20);
return w[20][20][20];
}else{
return w[20][20][20];
}
}else if(a<b&&b<c){
if(w[a][b][c-1]==inf){
w[a][b][c-1]=w_num(a,b,c-1);
}else
;
if(w[a][b-1][c-1]==inf){
w[a][b-1][c-1]=w_num(a,b-1,c-1);
}else
;
if(w[a][b-1][c]==inf){
w[a][b-1][c]=w_num(a,b-1,c);
}else
;
w[a][b][c]=w[a][b][c-1]+w[a][b-1][c-1]-w[a][b-1][c];
return w[a][b][c];
}else{
if(w[a-1][b][c]==inf){
w[a-1][b][c]=w_num(a-1,b,c);
}else
;
if(w[a-1][b-1][c]==inf){
w[a-1][b-1][c]=w_num(a-1,b-1,c);
}else
;
if(w[a-1][b][c-1]==inf){
w[a-1][b][c-1]=w_num(a-1,b,c-1);
}else
;
if(w[a-1][b-1][c-1]==inf){
w[a-1][b-1][c-1]=w_num(a-1,b-1,c-1);
}else
;
w[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 w[a][b][c];
}
}
int main(int argc, char *argv[]) {
long long int a,b,c,i,j,k;
for(i=0;i<21;i++)
for(j=0;j<21;j++)
for(k=0;k<21;k++)
w[i][j][k]=inf;
while(1){
scanf("%lld%lld%lld",&a,&b,&c);
if(a==-1&&b==-1&&c==-1)
break;
else{
if(a<=0||b<=0||c<=0){
printf("w(%lld, %lld, %lld) = %d\n",a,b,c,1);
continue;
}else if(a>20||b>20||c>20){
k=w_num(20,20,20);
}else if(a<b&&b<c){
k=w_num(a,b,c-1)+w_num(a,b-1,c-1)-w_num(a,b-1,c);
}else{
k=w_num(a-1,b,c)+w_num(a-1,b-1,c)+w_num(a-1,b,c-1)-w_num(a-1,b-1,c-1);
}
}
printf("w(%lld, %lld, %lld) = %lld\n",a,b,c,k);
}
return 0;
}