通常,正整数 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 的笨阶乘。
示例 1:
输入:4
输出:7
解释:7 = 4 * 3 / 2 + 1
示例 2:
输入:10
输出:12
解释:12 = 10 * 9 / 8 + 7 - 6 * 5 / 4 + 3 - 2 * 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/clumsy-factorial
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
# 若为乘除则计算,若为加减则入栈
class Solution:
def clumsy(self, N: int) -> int:
sign = 0 # 判断当前符号(0、1、2、3分别代表*、/、+、-)
stack = [N] # 初始化栈
for i in range(N - 1, 0, -1): # 依次遍历从N~1的各个元素
if sign == 0:
stack.append(stack.pop() * i) # 若为*则计算
elif sign == 1:
stack.append(int(stack.pop() / float(i))) # 若为/则计算
elif sign == 2:
stack.append(i) # 若为+则入栈
elif sign == 3:
stack.append(-i) # 若为-则入栈
sign = (sign + 1) % 4 # 计算出下次运算的符号
return sum(stack)