[题解] HDU 5528 Count a*b

[题解] HDU 5528 Count a*b 简单容斥+莫比乌斯反演

题意:令$f(m)$表示有多少个整数有序二元组$(a,b)$满足$0\le a,b<m$，且$m\nmid ab$。
输入$n$，求$g(n)={\sum }_{m|n}f(m)(mod{2}^{64})$ (也就是用$unsignedlonglong$)。$T$组数据，$1\le T\le 20000,1\le n\le 10^9$。
这道题对于其他莫比乌斯反演的题目的不同之处在于，此题借助实际意义来推式子，从而避免了大量运算，得到了极大的简化。这启示我们，遇到一道题，先尝试用意义去理解，而不要去傻推式子。
我们可以计算$m|ab$的二元组的个数，设为$h(m)$.那么
$$g(n)=\sum _{m|n}{m}^2-\sum _{m|n}h(m)$$
对于第一个函数，其等于
$$(1+p_1^2+p_1^4+\cdots +p_1^{2{\alpha }_1})\cdot \left(1+p_2^2+p_2^4+\cdots +p_2^{2{\alpha }_2}\right)\cdot \cdots \cdot (1+p_k^2+p_k^4+\cdots +p_k^{2{\alpha }_k})$$
对于每一个$n$，我们先预处理好$1$到$10^6$的素数，然后拿去作试除法(直接试除法时间复杂度$O(\sqrt{n})$会超时)，计算出幂次，然后直接算出每个$p$的贡献。
这里用了一种暴力统计积性函数卷积的新方法，值得注意。

对于第二个函数，记$d=\gcd (m,a)$，则$1=\gcd \left(\frac{m}{d},\frac{a}{d}\right)$
$$m\mid ab⟺\frac{m}{d}\mid \frac{a}{d}b⟺\frac{m}{d}\mid b$$
对于每一个$d$，$b$是$\frac{m}{d}$的倍数。于是我们可以找出$\frac{m}{\frac{m}{d}}=d$个$b$。
（这里我们把0看做$m$，不妨碍计数）

现在考虑如何生成$a$。对于每一个$d$，都有$\gcd \left(\frac{m}{d},\frac{a}{d}\right)=1$，其数量为$\phi \left(\frac{m}{d}\right)$。
因此$$h(m)=\sum _{d|m}d\phi \left(\frac{m}{d}\right)$$
而$$\begin{array}{rl}\sum _{m|n}h(m)& =\sum _{m|n}\sum _{d|m}d\phi \left(\frac{m}{d}\right)\\ & \underset{kd|n}{\overset{m=kd}{=}}\sum _{d|n}d\sum _{k|\frac{n}{d}}\phi \left(k\right)\\ & \stackrel{\phi \ast 1=id}{=}\sum _{d|n}d\cdot \frac{n}{d}\\ & =n\sum _{d|n}1\end{array}$$
我们依然采用素数试除法计算约数个数。
时间复杂度$O(10^6+T\pi \left(\sqrt{n}\right))$，其中$\pi (n)$为$n$的素数个数函数，
$\pi (n)\sim \frac{n}{\ln n}$，故时间复杂度为$O(10^6+T\frac{\sqrt{n}}{\log n})$。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#define pii pair<int,int>
using namespace std;
typedef unsigned long long ull;
const double eps = 1e-10;
const double pi = acos(-1.0);
const int maxn = 1e6 + 10;

int t;
int p[maxn],cnt;
bool isp[maxn];

void getp(){
	isp[1] = 1;
	for(int i = 2; i < maxn; i++){
		if(!isp[i]) p[++cnt] = i;
		for(int j = 1; j <= cnt && i * p[j] < maxn; j++){
			isp[i*p[j]] = 1;
			if(i%p[j] == 0) break;
		}
	}
}

inline ull qpow(ull a, int b){
	ull ans = 1, base = a;
	while(b){
		if(b&1) ans *= base;
		base *= base, b >>= 1;
	}
	return ans;
}

int n;

void solve(){
	getp();
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		int tmp = n;
		ull ans1 = 1, ans2 = n;
		for(int i = 1; i <= cnt && p[i]*p[i] <= n; i++){
			if(tmp%p[i] == 0){
				int res = 0;
				while(tmp%p[i] == 0){
					res++;
					tmp /= p[i];
				}
                ull tmpres = 1, base = p[i]*p[i];
                for(int j = 1; j <= res; j++){
					tmpres += base;
					base = base*(ull)p[i]*(ull)p[i];//不加ull会炸
				}
				//ans1 *= (qpow(p[i],2*res+2)-1)/((ull)p[i]*(ull)p[i]-1);
				//用快速幂会炸
				ans2 *= (res+1);
			}
		}
		if(tmp != 1) ans1 *= (1+(ull)tmp*(ull)tmp), ans2 *= 2;
		printf("%llu\n",ans1-ans2);
	}
}

int main()
{
	solve();
	return 0;
}

不得不说，这道题把常数卡到了极致(悲)