【发布时间】:2021-06-23 15:43:54
【问题描述】:
我正在实现一个计算第 n 个加泰罗尼亚数字的函数。序列公式如下:
我注意到记忆的解决方案比正常的递归解决方案慢。这是我的代码:
#include <bits/stdc++.h>
using namespace std;
int catalan_number_recursive(int n){
if (n == 0) return 1;
else{
int ans = 0;
for (int i = 0; i < n; i++){
ans += catalan_number_recursive(i)*catalan_number_recursive(n - 1 - i);
}
return ans;
}
}
int catalan_number_memo(int n, map<int, int> memo){
memo[0] = memo[1] = 1;
if (memo.count(n) != 0){
return memo[n];
}
else{
int ans = 0;
for (int i = 0; i < n; i++){
ans += catalan_number_memo(i, memo)*catalan_number_memo(n - 1 - i, memo);
}
memo[n] = ans;
return memo[n];
}
}
int main(){
printf("Catalan Numbers - DP\n\n");
int num = 12;
auto start1 = chrono::high_resolution_clock::now();
printf("%dth catalan number (recursive) is %d.\n", num, catalan_number_recursive(num));
auto finish1 = chrono::high_resolution_clock::now();
chrono::duration<double> elapsed1 = finish1 - start1;
cout << "Time taken: " << elapsed1.count() << "s.\n\n";
auto start2 = chrono::high_resolution_clock::now();
printf("%dth catalan number (memo) is %d.\n", num, catalan_number_memo(num, {}));
auto finish2 = chrono::high_resolution_clock::now();
chrono::duration<double> elapsed2 = finish2 - start2;
cout << "Time taken: " << elapsed2.count() << "s.\n";
return 0;
}
n = 12 的代码输出为:
Catalan Numbers - DP
12th catalan number (recursive) is 208012.
Time taken: 0.006998s.
12th catalan number (memo) is 208012.
Time taken: 0.213007s.
另外,当我尝试使用 n = 20 时,它给了我一个负值,这是不正确的,但对于较小的值,它是正确的。谢谢你的回答。
【问题讨论】:
-
您正在按值传递地图,这意味着每次调用都在制作副本,这很昂贵。而是通过参考传递并检查时间。
-
@NathanOliver 不仅如此,它仍然是指数级的。
标签: c++ recursion dynamic-programming memoization