[北理工乐学] NH-5 Probability

北理工乐学 NH-5 Probability

题目描述

在斗角场在你偷偷下了注，赌赢了，获得了一个小游戏资格，能免费送棒棒糖！

活动规则：在一个长度为$N$的数列的$N$个位置随机填入一个$\in [1,MAX]$的数，一个免费获得的棒棒糖数记为${\sum }_{i=1}^{N-1}[\gcd (a_i,a_{i+1})=a_i]$，$[\gcd (a_i,a_{i+1})=a_i]$表示如果$a_i$,$a_{i+1}$的最大公因数等于$a_i$，则$[\gcd (a_i,a_{i+1})=a_i]=1$，否则$[\gcd (a_i,a_{i+1})=a_i]=0$。富有的你不在意你能得到几个棒棒糖，而是想计算一下你随机填数后得到免费棒棒糖的期望数量。由于你很讨厌小数，你打算将答案对$10^9+7$取模。期望和取模的概念详见样例解释。

数据范围

$1\le T\le 6，1\le N\le 10^{12}，1\le MAX\le 10^{12}$，已知$MAX$不是P的倍数，$(P=10^9+7)$

题目大意

随机化一个长度为 $N$ 各个值在$[1,MAX]$的序列，求${\sum }_{i=1}^{N-1}[\gcd (a_i,a_{i+1})=a_i]$ 的期望。

分析

显然可以得到最终的答案为
$$\frac{{\sum }_{seq}{\sum }_{i=1}^{N-1}[\gcd (a_i,a_{i+1})=a_i]}{{MAX}^N}$$

分母可以利用快速幂求出逆元，所以我们的重点就变到了如何计算分子。

显然我们不可能枚举所有的序列以计算式子，所以我们考虑计算每个 $a_i$ 的贡献。

对于一个给定的 $a_i$，注意到要算它的贡献则必须把 $a_{i+1}$ 也确定下来，而整个序列其他位置可任意取值，所以一个给定的 $a_i$ 的贡献为 $MA{X}^{N-2}{\sum }_{a_{i+1}=1}^{MAX}[\gcd (a_i,a_{i+1})=a_i]$。而所有的 $a_i$ 均等价，所以分子就可化为：

$$(N-1)\cdot MA{X}^{N-2}\cdot \sum _{a_i=1}^{MAX}\sum _{k=1}^{MAX}[\gcd (a_i,k)=a_i]$$

接下来计算 ${\sum }_{k=1}^{MAX}[\gcd (a_i,k)=a_i]$：

这个式子的实际意义就是计算 $a_i$ 在 $[1,MAX]$ 上的倍数个数，显然可知这个数是 $\lfloor \frac{MAX}{a_i}\rfloor$，所以我们就仅需计算 ${\sum }_{a_i=1}^{MAX}\lfloor \frac{MAX}{a_i}\rfloor$。

但注意到 $MAX$ 值极大，我们无法直接计算这个和式，这里我选了一个比较取巧的方法（听大佬讲的）（不知道数论分块能不能做这道题，如有记得评论区评论一下）。

考虑计算 ${\sum }_{i=1}^{\lfloor \sqrt{MAX}\rfloor }\lfloor \frac{MAX}{i}\rfloor$，注意到这个是可以直接计算的，不妨认为答案即为 $2{\sum }_{i=1}^{\lfloor \sqrt{MAX}\rfloor }\lfloor \frac{MAX}{i}\rfloor$，因为在枚举因数的时候我们仅需考虑 $[1,\lfloor \sqrt{MAX}\rfloor ]$ 的部分，另外一部分可简单地通过除法来取得。

但直接计算会发现答案远远超出正确值，这也意味着我们必须找出重复值。

注意到对于 $i$，$\lfloor \frac{MAX}{i}\rfloor$ 和 $\lfloor \frac{i\sqrt{MAX}}{i}\rfloor$ 实际上是有重复的，这是必须扣除的。简单计算得到重复的数目为 $(\lfloor \sqrt{MAX}\rfloor {)}^2$

所以，本题答案为

$$\frac{(N-1)\cdot MA{X}^{N-2}\cdot (2\times {\sum }_{i=1}^{\lfloor \sqrt{MAX}\rfloor }-(\lfloor \sqrt{MAX}\rfloor {)}^2)}{MA{X}^N}$$

注意到本题数据极大，进行快速幂前应先对底数取模以保证数据不会炸 long long（我因为这个问题调了好久）。

参考代码

#include <stdio.h>
#include <math.h>

const long long Mod = (long long)1E9 + 7;

long long quickpow(long long a, long long n) {
    a %= Mod;
	long long ret = 1;
	while(n) {
		if(n & 1) ret = (ret * a) % Mod;
		a = (a * a) % Mod, n >>= 1;
	}
	return ret;
}

int main() {
	int T;
	scanf("%d", &T);
	while(T--) {
		long long N, M;
		scanf("%lld %lld", &N, &M);
		if(N == 1) {puts("0"); continue;}
		long long sum = 0;
		for(long long i = 1; i * i <= M; i++)
			sum = (sum + (M / i)) % Mod;
		sum = (sum * 2 % Mod - ((long long)(sqrt(M)) * (long long)(sqrt(M)) % Mod) + Mod) % Mod;
		sum = ((sum * ((N - 1) % Mod)) % Mod) * quickpow(M, N - 2) % Mod;
		printf("%lld\n", sum * quickpow(quickpow(M, N), Mod - 2) % Mod);
	}
	return 0;
}