首先,这是您的代码压缩(我这样做是为了回答,不要在实际代码中这样做。)
double function(int j, int i) {
if(i == 0 || j == 1) return 1;
if(i == 1 || j == 0) return j;
if(i > 0) return j * function(j, --i);
return 1 / (function(j, -i)); //changed this to -i
//might be a division by zero, you should check for that
}
由于最后一个块只能有效地发生在最外层的循环中,我们将把它拉出来:
double outer_function(int j, int i) {
if (i<0)
return 1 / inner_function(j, -i);
else
return inner_function(j, i);
}
double inner_function(int j, int i) {
if(i == 0 || j == 1) return 1;
if(i == 1 || j == 0) return j;
if(i > 0) return j * inner_function(j, --i);
}
我要做的第一件事是尝试将其放入尾递归形式。这涉及重新排列方程,因此递归之后没有任何内容。(我不是 100% 确定我做对了这一步)
double inner_function(int j, int i, int times=1) {
if(i == 0 || j == 1) return times;
if(i == 1 || j == 0) return times*j;
return inner_function(j, --i, times*j);
}
现在,因为在每个代码路径中,函数调用之后都没有代码,所以这是完全尾递归的。尾递归很容易变成迭代!
double inner_function(int j, int i, int times=1) {
while(true) {
if(i == 0 || j == 1) return times;
if(i == 1 || j == 0) return times*j;
//return inner_function(j, --i, times*j);
--i;
times *= j;
//go again!
}
}
如果我要从这里优化:
double function(int j, int i) {
bool invert = false;
if(i<0) {
i=-i;
invert=true;
}
double result=1;
if(i == 0) result = 0;
else if(j == 0) result = j;
else if (j != 1) {
while(i--)
result *= j;
}
return (invert ? 1/result : result);
}
或者,如果我猜你的意图:
double function(int j, int i) {
return std::pow(double(j), double(i));
}