[BZOJ 3309] DZY Loves Math

BZOJ传送门

Description

对于正整数$n$，定义$f(n)$为$n$所含质因子的最大幂指数。例如$f(1960)=f(2^3\ast 5^1\ast 7^2)=3,f(10007)=1,f(1)=0$。
给定正整数$a,b$，求$\sum \sum f(gcd(i,j))(i=1..a,j=1..b)$。

Input

第一行一个数$T$，表示询问数。
接下来$T$行，每行两个数$a,b$，表示一个询问。

Output

对于每一个询问，输出一行一个非负整数作为回答。

Sample Input

4
7558588 9653114
6514903 4451211
7425644 1189442
6335198 4957
 

Sample Output

35793453939901
14225956593420
4332838845846
15400094813

HINT

【数据规模】

$T\le 10000$

$1\le a,b\le 10^7$

解题分析

还是一波推式子：（不妨设$n\le m$）
$$\begin{gathered} \sum _{i=1}^n\sum _{j=1}^{m}f(gcd(i,j)) \\ =\sum _{d=1}^{n}f(d)\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{d}\rfloor }[gcd(i,j)=1] \\ =\sum _{d=1}^{n}f(d)\sum _{e=1}^{\lfloor \frac{n}{d}\rfloor }\mu (e)\lfloor \frac{n}{de}\rfloor \lfloor \frac{m}{de}\rfloor \end{gathered}$$
设$de=T$， 那么：
$$\begin{gathered} \sum _{d=1}^{n}f(d)\sum _{e=1}^{\lfloor \frac{n}{d}\rfloor }\mu (e)\lfloor \frac{n}{de}\rfloor \lfloor \frac{m}{de}\rfloor \\ =\sum _{T=1}^n\lfloor \frac{n}{T}\rfloor \lfloor \frac{m}{T}\rfloor \sum _{d|T}\mu (d)f(\frac{T}{d}) \end{gathered}$$
考虑如何计算$G(T)={\sum }_{d|T}\mu (d)f(\frac{T}{d})$。 显然我们要让$\mu (d)̸=0$， 从$T$内取得的相同质因数就不应该超过一个。

考虑$T=p_1^a\times ...\times p_n^a\times p_{n+1}^{b_1}\times ...\times p_{n+m}^{b_m}$， 其中$a>b_i$。

那么我们有两种选择：

• 将前$n$个最高次幂的选到$d$中， 后面随意选择， 这样得到的贡献是$(f(T)-1)\times (-1{)}^n\times {\sum }_{i=0}^m(\genfrac{}{}{0ex}{}{m}{i})(-1{)}^i$， 注意到后面当$m̸=0$的时候， 值为0， 当$m=0$的时候， 值为$(f(T)-1)\times (-1{)}^n$。
• 不选完前$n$个， 后面随意选择， 这样得到的贡献是$f(T)\times {\sum }_{i=0}^{n-1}\left(\genfrac{}{}{0ex}{}{n}{i}\right)\times (-1{)}^n\times {\sum }_{j=0}^m(\genfrac{}{}{0ex}{}{m}{j})(-1{)}^j$， 同理当$m̸=0$的时候值为$0$， 当$m=0$的时候， 值为$f(T)\times (-1{)}^{(n+1)}$。

综上， $G(T)$当且仅当$T$的质因数的指数齐次的时候有值， 值为$(-1{)}^{n+1}$。

所以我们可以暴力筛出所有质数， 然后$DFS$或筛出得到一次齐次的所有数， 然后向上累乘即可。 总复杂度$O(N+T\sqrt{N})$。

代码如下：

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <algorithm>
#include <cstring>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define MX 10000005
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
int tim[MX], num[MX], pri[MX], sum[MX], buc[MX];
bool npr[MX], vis[MX];
int pcnt, tot;
void get_pri()
{
	R int i, j, tar;
	for (i = 2; i <= 1e7; ++i)
	{
		if (!npr[i]) pri[++pcnt] = i, num[i] = 1;
		for (j = 1; j <= pcnt; ++j)
		{
			tar = pri[j] * i;
			if (tar > 1e7) break;
			npr[tar] = true;
			if (!(i % pri[j])) break;
		}
	}
}
void DFS(R int pos, R int now, R int dep)
{
	buc[++tot] = now, num[now] = dep;
	long long tar;
	for (R int i = pos + 1; i <= pcnt; ++i)
	{
		tar = 1ll * now * pri[i];
		if (tar > 1e7) break;
		DFS(i, tar, dep + 1);
	}
}
int main(void)
{
	get_pri();
	DFS(0, 1, 0); long long mul;
	for (R int i = 2; i <= tot; ++i)
	{
		for (mul = 1ll * buc[i] * buc[i]; mul <= 1e7; mul = mul * buc[i])
		num[mul] = num[buc[i]];
	}
	for (R int i = 1; i <= 1e7; ++i)
	{
		if (!num[i]) sum[i] = sum[i - 1];
		else sum[i] = sum[i - 1] + ((num[i] & 1) ? 1 : -1);
	}
	int T, n, m, lef, rig; in(T); long long ans;
	W (T--)
	{
		in(n), in(m); ans = 0;
		if (n > m) std::swap(n, m);
		for (lef = 1; lef <= n; lef = rig + 1)
		{
			rig = std::min(n / (n / lef), m / (m / lef));
			ans += 1ll * (n / lef) * (m / lef) * (sum[rig] - sum[lef - 1]);
		}
		printf("%lld\n", ans);
	}
}