每日算法系列【LeetCode 1006】笨阶乘

关注公众号【算法码上来】,每日算法干货马上就来!

题目描述

通常,正整数 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

提示

  • $1 \le N \le 10000$
  • $-2^{31} \le answer \le 2^{31} - 1$ (答案保证符合 32 位整数。)

题解

笨方法我就不写了,按照运算顺序模拟一遍,数量比较少的话,应该也不会超时。

这道题可以用简单的数学计算来优化一下,首先笨阶乘数学形式是下面这样的:
$$
clumsy(n) = \left\lfloor\frac{n(n-1)}{n-2}\right\rfloor + (n-3) - \left\lfloor\frac{(n-4)(n-5)}{n-6}\right\rfloor + \cdots
$$

好像看不出什么东西,那么我们先去掉取整符号看看:
$$
\left\lfloor\frac{n(n-1)}{n-2}\right\rfloor = \left\lfloor\frac{(n-2)(n-2)+3(n-2)+2}{n-2}\right\rfloor = \left\lfloor n+1+\frac{2}{n-2}\right\rfloor
$$

所以当 $n \ge 5$ 时,取整符号可以直接去掉:
$$
\left\lfloor\frac{n(n-1)}{n-2}\right\rfloor = n + 1
$$

所以我们就讨论 $n \ge 5$ 时的情况,笨函数可以写成:
$$
clumsy(n) = n+1 + n-3 - (n-3) + \cdots = n+1 + d(n)
$$

也就是第一个取整($3$ 项)结果是 $n+1$ ,而后面每 $4$ 项结果是 0 ,最后还剩余不足 $4$ 项的结果 $d(n)$ 。

那么 $d(n)$ 的取值有 $4$ 种情况:

  • $n \% 4 = 0$ ,也就是最后剩余 $1$ 项时,$d(n) = 1$ 。
  • $n \% 4 = 1$ ,也就是最后剩余 $2$ 项时,$d(n) = 2 - 1 = 1$ 。
  • $n \% 4 = 2$ ,也就是最后剩余 $3$ 项时,$d(n) = 3 - 2 \times 1 = 1$ 。
  • $n \% 4 = 3$ ,也就是最后剩余 $0$ 项时,$d(n) = 0$ 。

但是考虑到取整函数只有在 $n \ge 5$ 时才等于 $n+1$ ,也就是最后剩余的项数大于 $2$ 时才行。那么上面的 $4$ 种情况的 $1、4$ 两种,就要特殊处理一下前一个取整结果:

  • $n \% 4 = 0$ ,也就是最后剩余 $1$ 项时,$d(n) = 5 - 4 \times 3 / 2 + 1 = 0$ 。
  • $n \% 4 = 1$ ,也就是最后剩余 $2$ 项时,$d(n) = 2 - 1 = 1$ 。
  • $n \% 4 = 2$ ,也就是最后剩余 $3$ 项时,$d(n) = 3 - 2 \times 1 = 1$ 。
  • $n \% 4 = 3$ ,也就是最后剩余 $0$ 项时,$d(n) = 4 - 3 \times 2 / 1 = -2$ 。

综上,$n < 5$ 时的结果直接算出来就行了,$n \ge 5$ 时结果就是 $n+1+d(n)$ 。

代码

c++

class Solution {
public:
    int clumsy(int N) {
        int q = N & 3;
        int a[5] = {0, 1, 2, 6, 7};
        int d[4] = {1, 2, 2, -1};
        return N>=5 ? N+d[q] : a[N];
    }
};

python

class Solution:
    def clumsy(self, N: int) -> int:
        q = N & 3
        a = [0, 1, 2, 6, 7]
        d = [1, 2, 2, -1]
        return N+d[q] if N>=5 else a[N]

   转载规则


《每日算法系列【LeetCode 1006】笨阶乘》 韦阳 采用 知识共享署名 4.0 国际许可协议 进行许可。
 上一篇
每日算法系列【LeetCode 684】冗余连接 每日算法系列【LeetCode 684】冗余连接
关注公众号【算法码上来】,每日算法干货马上就来! 题目描述在本问题中, 树指的是一个连通且无环的无向图。 输入一个图,该图由一个有着 $N$ 个节点(节点值不重复 $1, 2, …, N$)的树及一条附加的边构成。附加的边的两个顶点包
2020-02-21
下一篇 
每日算法系列【LeetCode 287】寻找重复数 每日算法系列【LeetCode 287】寻找重复数
关注公众号【算法码上来】,每日算法干货马上就来! 题目描述给定一个包含 $n + 1$ 个整数的数组 $nums$,其数字都在 $1$ 到 $n$ 之间(包括 $1$ 和 $n$),可知至少存在一个重复的整数。假设只有一个重复的整数,
2020-02-19
  目录