HDU 6069 Counting Divisors （2017 Multi-Univ Training Contest 4）

Problem

定义 d(n) 表示数字 n 的约数个数

给定 l, r, k, 求 $(\sum_{i=l}^{r}{d(i^k)}) \mod{998244353}$

Limit

$1\le l \le r \le10^{12}$

$1 \le k \le 10^7$

Idea

$n = a_1^{p_1}\cdot a_2^{p_2}\cdots a_i^{p_i}$ （其中 $a_i$ 均为质数）

根据组合原理求 n 的约数个数 : $d(n) = (p_1+1)(p_2+1) \cdots (p_i+1)$

由于 l 与 r 最大 $10^{12}$ ，故预处理出 [1, 1e6] 的所有质数。则 [l, r] 中每个数最多只有一个比 1e6 大的质数。（若存在两个及以上的大于 1e6 的质数，则乘积必然大于 $10^{12}$ 。

枚举 [1, 1e6] 的每个质数，对于每个质数 prime 枚举区间 [l, r] 的巧合为 prime 整数倍的数 number ，循环处理 number 有多少个 prime 因子存在。假设为 cnt 个，则该 $number^k$ 含有 $cnt\times k$ 个质数，对 $d(number)$ 贡献为 $\times (cnt\times k + 1)$ 。

对 [l, r] 的 d(number) 求和。

应注意可能含有一个比 1e6 的质数，此处因进行处理。

Code

#include<bits/stdc++.h>
using namespace std;
const int mod = 998244353;
const int N = 1000000 + 10;
bool isprime[N];
int primes[N/3], tot, T, k;
long long divs[N], l, r, arr[N];
void getPrime() {
    tot = 0;
    memset(isprime, 0, sizeof(isprime));
    for(int i=2, j;i<N;i++) {
        if(!isprime[i]) primes[tot++] = i;
        for(j=0;j<tot && i*primes[j] < N;++j) {
            isprime[i*primes[j]] = true;
            if(i % primes[j] == 0)  break;
        }
    }
}
int main()
{
    getPrime();
    scanf("%d", &T);
    while(T-- && scanf("%lld %lld %d", &l, &r, &k)!=EOF)
    {
        for(long long i=l;i<=r;i++)
        {
            arr[i-l] = i;
            divs[i-l] = 1;
        }
        for(int i=0;i<tot;i++)
        {
            long long start = l/primes[i]*primes[i] + (l%primes[i]?primes[i]:0);
            for(long long j=start, cnt;j<=r;j+=primes[i])
            {
                cnt = 0;
                while(arr[j-l] % primes[i] == 0) {
                    cnt++;
                    arr[j-l] /= primes[i];
                }
                (divs[j-l] *= (cnt * k + 1ll)) %= mod;
            }
        }
        long long ans = 0;
        for(long long i=l;i<=r;i++)
        {
            if(arr[i-l] != 1)    (divs[i-l] *= (k+1ll)) %= mod;
            (ans += divs[i-l]) %= mod;
        }
        printf("%lld\n", ans);
    }
}