Regional 2015 - Asia Changchun - B Count a × b_count a脳bicpc 2015 changchun-CSDN博客

本文链接：https://blog.csdn.net/skywalkert/article/details/49276505

题目：http://acm.nenu.edu.cn/regional-2015/files/2015/10/problemset_20151015.pdf

题意：
令 $f(m)$ 表示满足 $0\le a,b<m$ 且 $m∤ab$ 的二元组 $(a,b)$ 的个数。
令 $g(n)=\sum_{m|n}{f(m)}$ ，询问 $T$ 次某个 $g(n)$ 的值，答案模 $2^{64}$ 。
$T\le 20000, n \le 10^9$ 。

题解：
（64位无符号整型变量的运算可以看做是在模 $2^{64}$ 意义下的运算）
看到 $g(n)$ 的形式很容易想到一个事实，如果 $f(m)$ 是积性函数，即”对于互质的 $a$ 和 $b$ ，有 $f(ab)=f(a)\cdot f(b)$ ”，那么 $g(n)$ 的表达式可以做到如下的改写：
令 $n=p_1^{k_1}\cdot p_2^{k_2}\cdots p_t^{k_t}$ ，那么

g (n) = \sum m | n f (m) = (f (1) + f (p 1) + f (p 21) + \dots + f (p k 1 1)) \cdot (f (1) + f (p 2) + f (p 22) + \dots + f (p k 2 2)) \dots (f (1) + f (p t) + f (p 2 t) + \dots + f (p k t t)) = \prod i = 1 t \sum j = 0 k i f (p j i)

$g(n)=\sum_{m|n}{f(m)}\\=(f(1)+f(p_1)+f(p_1^2)+\cdots+f(p_1^{k_1}))\\\cdot(f(1)+f(p_2)+f(p_2^2)+\cdots+f(p_2^{k_2}))\\\cdots(f(1)+f(p_t)+f(p_t^2)+\cdots+f(p_t^{k_t}))\\=\prod_{i=1}^{t}{\sum_{j=0}^{k_i}{f(p_i^j)}}$
上面式子可以这样理解，对于每个

n $n$ 的因子

m $m$ ，

m $m$ 对应的质因子的次幂一定小于等于

n $n$ 对应的质因子次幂，从上述乘积的每一个括号里选出一个数字，乘起来就对应一个

f(m) $f(m)$ ，这样算是不重不漏的。
这样可以将

O(n−−√) $O(\sqrt{n})$ 枚举因子的过程转化为

O(logn) $O(logn)$ 枚举质因子的过程。
然而这个题里的

f(m) $f(m)$ 并不积性，原因很容易想到，因为它的定义应该是一个积性函数的反面，那么我们来看看这个反面。
定义

h(m) $h(m)$ 表示满足

0≤a,b<m $0\le a,b<m$ 且

m|ab $m|ab$ 的二元组

(a,b) $(a,b)$ 的个数，那么显然就有

f(m)=m2−h(m) $f(m)=m^2-h(m)$ ，其中

m2 $m^2$ 显然是一个积性函数，而考虑计算

h(m) $h(m)$ 的时候，不难发现

a $a$ 和

b $b$ 里不同的质因子之间不会产生影响，可以分别把每种质因子的情况算出来之后利用乘法原理乘起来，这也就意味着

h(m) $h(m)$ 是积性的，所以如果知道

h(pk) $h(p^k)$ 怎么算，就可以利用上述的转化过程

O(logn) $O(logn)$ 求解。
先考虑

n $n$ 的所有因子

m $m$ 对应的

m2 $m^2$ 之和，直接按照上式计算即可。
再来考虑

h(m) $h(m)$ ，显然

a $a$ 和

b $b$ 取值范围从

0≤a,b<m $0\le a,b<m$ 变成

1≤a,b≤m $1\le a,b\le m$ 是不会改变答案的。
考虑

m=pk $m=p^k$ 的情况，令

a=pta⋅ra,b=ptb⋅rb $a=p^{t_a}\cdot r_a, b=p^{t_b}\cdot r_b$ ，则

ra≤pk−ta且(ra,p)=1 $r_a\le p^{k-t_a}且(r_a,p)=1$ ，

rb≤pk−tb且(rb,p)=1 $r_b\le p^{k-t_b}且(r_b,p)=1$ 。实际上，如果固定了

ta $t_a$ 和

tb $t_b$ ，

(a,b) $(a,b)$ 的个数是可以直接算出来的，即

ϕ(pk−ta)⋅ϕ(pk−tb) $\phi(p^{k-t_a})\cdot\phi(p^{k-t_b})$ 。
现在所求即

h (m) = h (p k) = \sum t a = 0 k \sum t b = 0 k [t a + t b \geq k] \cdot ϕ (p k - t a) \cdot ϕ (p k - t b) = \sum t a = 0 k \sum t b = 0 k [t a + t b \leq k] \cdot ϕ (p t a) \cdot ϕ (p t b) = \sum i = 0 k \sum j = 0 i ϕ (p j) \cdot ϕ (p k - j) = 1 + \sum i = 1 k (i + 1) p i - 2 i p i - 1 + (i - 1) p i - 2 = (k + 1) p k - k p k - 1

$h(m)=h(p^k)=\sum_{t_a=0}^{k}\sum_{t_b=0}^{k}{[t_a+t_b\ge k]\cdot\phi(p^{k-t_a})\cdot\phi(p^{k-t_b})}\\=\sum_{t_a=0}^{k}\sum_{t_b=0}^{k}{[t_a+t_b\le k]\cdot\phi(p^{t_a})\cdot\phi(p^{t_b})}\\=\sum_{i=0}^{k}\sum_{j=0}^{i}{\phi(p^j)\cdot\phi(p^{k-j})}\\=1+\sum_{i=1}^{k}{(i+1)p^i-2ip^{i-1}+(i-1)p^{i-2}}\\=(k+1)p^k-kp^{k-1}$
更进一步地，有

∑ki=0h(pi)=(k+1)pk $\sum_{i=0}^{k}{h(p^i)}=(k+1)p^k$ 。
于是可以预处理出

O(n−−√) $O(\sqrt{n})$ 以内的质数，单次

O(logn) $O(logn)$ 计算答案。

代码：

#include <cstdio>
typedef unsigned long long ULL;
const int maxn = 32000;
int tot, prime[maxn], t, n;
bool vis[maxn];
ULL ans1, ans2;
int main()
{
    for(int i = 2; i < maxn; ++i)
    {
        if(!vis[i])
            prime[tot++] = i;
        for(int j = 0, o; j < tot && (o = i * prime[j]) < maxn; ++j)
        {
            vis[o] = 1;
            if(i % prime[j] == 0)
                break;
        }
    }
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        ans1 = ans2 = 1;
        for(int i = 0; i < tot && prime[i] * prime[i] <= n; ++i)
            if(n % prime[i] == 0)
            {
                int cnt = 0, tmp = 1;
                ULL sum = 1;
                for( ; n % prime[i] == 0; n /= prime[i], ++cnt);
                for(int j = 1; j <= cnt; ++j)
                {
                    tmp *= prime[i];
                    sum += (ULL)tmp * tmp;
                }
                ans1 *= sum;
                ans2 *= tmp * (cnt + 1ULL);
            }
        if(n > 1)
        {
            ans1 *= (ULL)n * n + 1;
            ans2 *= n * 2ULL;
        }
        printf("%llu\n", ans1 - ans2);
    }
    return 0;
}