洛谷P3935 Calculating 题解

洛谷P3935 Calculating 题解

题目链接：P3935 Calculating

题意：

若 $n$ 的分解质因数结果为 ${\prod }_{i=1}^s{p}_i^{k_i}$

令 $f(n)={\prod }_{i=1}^s(k_i+1)$

求
$$\sum _{i=l}^{r}f(i)\mathrm{m}\mathrm{o}\mathrm{d}998244353$$

根据容斥原理，可以把答案分为两个部分
$$\sum _{i=l}^{r}f(i)=\sum _{i=1}^{r}f(i)-\sum _{i=1}^{l-1}f(i)$$
然后推推柿子
$$\begin{array}{rl}\sum _{i=1}^{r}f(i)& =\sum _{i=1}^r\prod _{j=1}^{s_i}(k_{ij}+1)\\ & =\sum _{i=1}^r\sum _{d\mid i}1\\ & =\sum _{i=1}^r\lfloor \frac{r}{i}\rfloor \end{array}$$
于是有
$$\sum _{i=1}^{n}f(i)=\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$$
考虑数论分块

时间复杂度 $O(\sqrt{r})$

代码如下

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define INF 0x3f3f3f3f3f3f3f3f
const int mod=998244353;
int L,R;
int solve(int n)
{
    int l=1,r=0,res=0;
    while(l<=n)
    {
        r=n/(n/l);
        res=(res+(r-l+1)*(n/l)%mod)%mod;
        l=r+1;
    }
    return res%mod;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    // freopen("check.in","r",stdin);
    // freopen("check.out","w",stdout);
    cin >> L >> R;
    cout << (solve(R)-solve(L-1)+mod)%mod << endl;
    return 0;
}

转载请说明出处