笨阶乘
问题:
通常,正整数 n 的阶乘是所有小于或等于 n 的正整数的乘积。例如,factorial(10) = 10 * 9 * 8 * 7 * 6 * 5 * 4 * 3 * 2 * 1。
相反,我们设计了一个笨阶乘 clumsy:在整数的递减序列中,我们以一个固定顺序的操作符序列来依次替换原有的乘法操作符:乘法(*),除法(/),加法(+)和减法(-)。
例如,clumsy(10) = 10 * 9 / 8 + 7 - 6 * 5 / 4 + 3 - 2 * 1。然而,这些运算仍然使用通常的算术运算顺序:我们在任何加、减步骤之前执行所有的乘法和除法步骤,并且按从左到右处理乘法和除法步骤。
另外,我们使用的除法是地板除法(floor division),所以 10 * 9 / 8 等于 11。这保证结果是一个整数。
实现上面定义的笨函数:给定一个整数 N,它返回 N 的笨阶乘。
思路:
方法一,算法优化。
经过数学分析计算。参考证明
- 当 N % 4 == 0 时,N 的笨阶乘为 N + 1。
- 当 N % 4 == 3 时,N 的笨阶乘为 N - 1。
- 当 N % 4 == 1 或 2 时,N 的笨阶乘为 N + 2。
- 当 N 小于 5 时单独计算。其他均满足上述规律。
class Solution {
public:
int clumsy(int N) {
switch(N) {
case 1 :
return 1;
case 2 :
return 2;
case 3 :
return 6;
case 4 :
return 7;
default : break;
}
switch(N % 4) {
case 0 :
return N + 1;
case 3 :
return N - 1;
default :
return N + 2;
}
}
};
时间复杂度:O(1)
空间复杂度:O(1)
方法二:
- 用一个栈临时存储 笨阶乘 产生的所有的数的和的数字。
- 按照 *、/、+、- 的顺序计算每一个和并压入栈中。
- 将栈中元素求和即为笨阶乘。
class Solution {
public:
int clumsy(int N) {
stack<int> stk;
stk.push(N);
N--;
auto index = 0; // 用于控制乘、除、加、减
while (N > 0) {
switch(index % 4) {
case 0 :
stk.top() *= N; break;
case 1 :
stk.top() /= N; break;
case 2:
stk.push(N); break;
case 3:
stk.push(-N); break;
default : break;
}
index++;
N--;
}
// 把栈中所有的数字依次弹出求和
auto sum = 0;
while (!stk.empty()) {
sum += stk.top();
stk.pop();
}
return sum;
}
};
时间复杂度:O(n)
空间复杂度:O(n)