NC655 牛牛算题

题目陈述

大意：给定一个正整数$n$，充当被除数。对于$p,1\le p\le n$，充当除数，求所有$p$的商与余数的乘积之和，即$\sum k\ast m$，答案取模$mod$

算法一：朴素算法

算法思路

• 根据题意，一个很显然的算法，就是对于每一个$p$分别计算出他的$k,m$然后计算$k\ast m$

代码实现

class Solution {
public:
    long long cowModCount(long long n) {
        long long ans = 0, mod = 1e9 + 7, k, m;
        for (int p = 1; p <= n; p ++ ) //枚举所有合法的p
        {
            k = n / p; //求解商
            m = n % p; //求解余数
            ans = (ans + k * m % mod) % mod; //将他们的贡献加入到答案中
        }
        return ans;
    }
};

复杂度分析

• 时间复杂度，每个合法的$p$都遍历了一遍，即为$p$的区间大小，时间复杂度为$O(n)$，当然$n_{max}=INT\_MAX$，显然还是会$TLE$的
• 空间复杂度，定义了四个变量$ans,mod,k,m$，为$O(1)$

算法二：数学推导+整除分块

前置知识：整除分块

• 此处我们我们需要先了解一个前置知识：整除分块。

• 首先，要了解整除分块是干什么的？设计整除分块的题目大概都具有以下形式$$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$$

• 上面的括号代表向下取整，即整除

• 我们以$n=6$为例子
  
  

• 我们可以发现对于一个$i$，往往是一段区间的整除值都是同一个值（废话）

• 假如我们通常知道了区间内的一个下标（一般是左端点），如何求得跟它所在区间的右端点？

• 设$n=i\ast j+r，r\in [0,i-1]$，显然我们可以这样表示$i$，而把$j$设置为倍数

• 对于右端点$r$,必然有$r<j$

• 反证法，若$r>=j$则，可以令$r^{\prime }=r-j$,$n=r\ast j+r=(r+1)\ast j+r^{\prime }$

• 就得到了区间内比右端$r$还靠右的满足条件的点

• 所以：对于右端点$r$,必然有$r<j$

• 我们就可以根据这个性质，通过$l$，求得$j$，再通过$j$求得$r$

• 故对于一个区间，我们知道左端点$l$，即可以通过公式计算右端点$r=\lfloor \frac{n}{\lfloor \frac{n}{l}\rfloor }\rfloor$，代码中写作r = n / (n / i)，注意此处为整除

• 时间复杂度为$O(\sqrt{n})$

算法思路

• 显然这是一道数学题，我们考虑数学方法，当然，也少不了推导和化简
• 对于所有的$p,(p\in [1,n])$
• 有$$ans=\sum \lfloor \frac{n}{p}\rfloor \times (n\%p)$$
• 即
  $$ans=\sum _{p=1}^n\lfloor \frac{n}{p}\rfloor \times (n-p\times \lfloor \frac{n}{p}\rfloor )$$
• 整除分块思想有，$k\in [l,r]$,其中$r=\lfloor \frac{n}{\lfloor \frac{n}{l}\rfloor }\rfloor$
• 同一块$\lfloor \frac{n}{k}\rfloor$都相同，即$\lfloor \frac{n}{k}\rfloor =\lfloor \frac{n}{l}\rfloor$
• 故对于同一块有$$\sum _{k=l}^r\lfloor \frac{n}{l}\rfloor \times (n-k\times \lfloor \frac{n}{l}\rfloor )$$
• 展开括号得
  $$\sum _{k=l}^r(n\times \lfloor \frac{n}{l}\rfloor -k\times \lfloor \frac{n}{l}{\rfloor }^2)$$
• 求和符号性质有
  $$\sum _{k=l}^{r}n\times \lfloor \frac{n}{l}\rfloor -\sum _{k=l}^{r}k\times \lfloor \frac{n}{l}{\rfloor }^2$$
• 令$len=r-l+1$
• 化简求和结果，等差数列公式有
  $$len\times n\times \lfloor \frac{n}{l}\rfloor -\frac{(l+r)\times len}{2}\times k\times \lfloor \frac{n}{l}{\rfloor }^2$$
• 现在已知每一块的$O(1)$求法
• 整除分块得到了每一块的答案即可求得$ans$
• 下面是我的手写推导过程
  

关于不用逆元

• 肯定有同学好奇为什么我代码里面除了$2$，但是没有用逆元？
• 因为那一整项求和公式，必然能整除2，故不需要逆元
• 证明如下：
• 只要$(l+r)$和$len$中有一个为偶数，就可以整除2
• $l,r$奇偶性相同时候，$(l+r)$为偶数，$len=r-l+1$为奇数，能整除
• $l,r$奇偶性不同的时候,$(l+r)$为奇数，$len=r-l+1$为偶数，能整除
• 故此处不需要用到逆元

复杂度分析

• 时间复杂度，整除分块为$O(\sqrt{n})$，对于每一块表达式求值为$O(1)$，总得时间复杂度为$O(\sqrt{n})$
• 空间复杂度，定义了几个变量，为$O(1)$

代码实现

class Solution {
public:
    long long cowModCount(long long n) {
        long long ans = 0, mod = 1e9 + 7;
        for (long long l = 1, r, len, tmp; l <= n; l = r + 1)
        {
            tmp = n / l;//重复计算所以用tmp暂存
            r = n / tmp;//根据公式计算右边界
            len = r - l + 1;//计算区间长度
            ans = (ans + len * tmp * n % mod - tmp * tmp % mod * (l + r) * len / 2 % mod) % mod;
        }//根据我们上面计算的公式，累加到ans值上面
        if (ans < 0)
            ans += mod;//C++里面负数取模依旧为负数，需要加上mod
        return ans;
    }
};