【例题3】这个问题直接给出了一段求函数w(a, b, c)的伪代码:
def w(a,b,c):
if a<=0 or b<=0 or c<=0:
return 1
elif a>20 or b>20 or c>20:
return w(20,20,20)
elif a<b and b<c:
return w(a,b,c-1) + w(a,b-1,c-1) - w(a,b-1,c)
else:
return w(a-1,b,c)+w(a-1,b-1,c)+w(a-1,b,c-1)
要求给定a, b, c,求w(a, b, c)的值。
看似代码使用递归,但是细想的话,时间复杂度为O(3^n),计算量非常庞大。
用动态规划思想解题,时间复杂度为O(n^3):
def w(a,b,c):
if a>20 or b>20 or c>20:
a, b, c = 20, 20, 20
db = [[[0 for k in range(c+1)] for j in range(b+1)] for i in range(a+1)]
for i in range(a+1):
for j in range(b+1):
for k in range(c+1):
if i*j*k == 0:
db[i][j][k] = 1
elif a<b<c:
db[i][j][k] = db[i][j][k-1] + db[i][j-1][k-1] - db[i][j-1][k]
else:
db[i][j][k] = db[i-1][j][k] + db[i-1][j-1][k] + db[i-1][j][k-1]
return db[-1][-1][-1]