我是C++的新手,但我认为解决一些欧拉计划问题会让我熟悉这门语言。
当尝试解决欧拉计划第14题: 最长考拉兹序列时,我的C++解法无法正常工作,但是我的Python解法没有问题...
import time
start = time.time()
memo = {1:1,2:2}
longest_chain, longest_starting_key = 2, 2
def rec(a):
global longest_chain, longest_starting_key
if a in memo.keys():
return memo[a]
if a % 2 == 0:
memo[a] = rec(a // 2) + 1
else:
memo[a] = rec(3 * a + 1) + 1
if memo[a] > longest_chain:
longest_chain = memo[a]
longest_starting_key = a
return memo[a]
for i in range(1000000,3,-1): rec(i)
print("starting key", longest_starting_key , ": has length", longest_chain)
print((time.time() - start), "seconds")
并获得
starting key 837799 has length 525
1.399820327758789 seconds
我的C++尝试...(我认为它是等价的...)
#include <iostream>
#include <unordered_map>
std::unordered_map<int, int> mem = { {1,1},{2,2} };
int longest_chain{ 2 }, best_starting_num{ 2 };
bool is_in(const int& i){
auto search = mem.find(i);
if(search == mem.end())
return false;
else
return true;
}
int f(int n){
if(is_in(n))
return mem[n];
if(n % 2)
mem[n] = f(3 * n + 1) + 1;
else
mem[n] = f(n / 2) + 1;
if(mem[n] > longest_chain)
longest_chain = mem[n];
return mem[n];
}
int main(){
for(int i = 3; i < 1000000; i++){
f(i);
}
std::cout << longest_chain << std::endl;
}
在Visual Studio的调试模式下运行时,我遇到了以下错误。
Unhandled exception at 0x00007FF75A64EB70 in
0014_longest)collatz_sequence_cpp.exe: 0xC00000FD: Stack overflow
(parameters: 0x0000000000000001, 0x000000DC81493FE0).
有人告诉我应该使用指针在堆上进行分配,但我对指针的使用经验非常有限...
另一件我不理解的事情是...当我在主循环中使用100'000而不是1'000'000时,它可以运行,但速度非常慢。
mem
已经在堆中分配了新元素。 - Jean-François Fabre