记忆数组—递归算法的优化
原理:
以斐波那契数列为例子
递归 + 记忆化 约= 递推
- 递推
#include <iostream>
using namespace std;
int main() {
int x;
long long num[55] = {0, 1, 1};
cin >> x;
for (int i = 3; i <= x; i++) {
num[i] = num[i - 1] + num[i - 2];
}
cout << num[x] << endl;
return 0;
}
- 递归
#include <iostream>
using namespace std;
long long func(int x) {
if (x == 1 || x == 2) return 1;
return func(x - 1) + func(x - 2);
}
int main() {
int x;
cin >> x;
cout << func(x) << endl;
return 0;
}
这个程序与上面的递推函数写法相比,比如计算func(50)的值时, 会分为49和48的计算,48分为47、46的计算, 49分为48、47的计算,最后在计算过程中,会出现极多的重复的计算,当数据过大时,会导致时间的大量消耗,甚至超过规定时间。
----记忆化数组的原理-----
这首时候我们可以使用一个数组,当一个数被计算过的时候,我们就将这个数存储在这个记忆化数组中,我们在递归执行的过程中,判断是否计算过,如果计算过,直接返回这个数的值就可以使用了,这样子就避免了重复的计算。
下面是递归的优化(记忆数组)
#include <iostream>
using namespace std;
long long num[55];//记忆数组
long long func(int x) {
if (x == 1 || x == 2) return 1;
if (num[x] ! = 0) {//优化,num[x]不等于0,证明计算过func(x)并存在了num数组中
return num[x];
}
return num[x] = func(x - 1) + func(x - 2);//借助赋值表达式,计算f(x)并存为num[x]
//逗号表达式: cout << (3, 4, 5) << endl; ----> 5 //返回值是最后一个逗号后面的值
}
int main() {
int x;
cin >> x;
cout << func(x) << endl;
return 0;
}
例题euler14
#include<iostream>
using namespace std;
int num[10000000]; //记忆化数组
int func(long long x) {
if (x == 1) return 1;
if ( x < 10000000 && num[x] != 0 ){//表明呗计算过了
return num[x];//直接返回这个数使用
}
int t = 0;
if (x % 2 == 0) t = func(x / 2) + 1;
else t = func(3 * x + 1) + 1;
if ( x < 10000000 ) {
num[x] = t; //将这个计算过的数字存到num数组中
}
return t;
}
int main() {
int ans = 1;
for (int i = 1; i <= 1000000; i++) {
if (func(ans) < func(i)){
ans = i;
}
}
cout << ans << endl;
return 0;
}
写于2021-07-08