LG P4449 & JZOJ 于神之怒

( ext{Problem})

JZOJ上，求

[sum_{i=1}^n sum_{j=1}^m gcd(i,j)^k ]

对 (10^9+7) 取模
(n,m,k le 5 imes 10^6)

LG 上，是一个加强版，有 (T（Tle 2 imes 10^3）) 组数据

( ext{Analysis})

依照套路的方法，我们可以推出

[Ans = sum_{p=1} p^k sum_{d=1} mu(d) lfloor frac{n}{pd} floor lfloor frac{m}{pd} floor ]

若只有一组数据，那么
数论分块套数论分块 (O(n^{frac{3}{4}})) 即可
加上线筛 (O(n))

( ext{Code})

#include<cstdio>
#include<iostream>
#define LL long long
#define re register
using namespace std;

const int N = 5e6, P = 1e9 + 7;
int n, m, k, totp, pr[N], vis[N + 5], sum[N + 5], pk[N + 5];

inline int fpow(LL x, LL y)
{
	LL res = 1;
	for(; y; y >>= 1)
	{
		if (y & 1) res = res * x % P;
		x = x * x % P;
	}
	return res;
}

inline void Euler()
{
	vis[1] = sum[1] = pk[1] = 1;
	for(re int i = 2; i <= N; i++)
	{
		if (!vis[i]) pr[++totp] = i, sum[i] = -1, pk[i] = fpow(i, k);
		for(re int j = 1; j <= totp && i * pr[j] <= N; j++)
		{
			vis[i * pr[j]] = 1, pk[i * pr[j]] = (LL)pk[i] * pk[pr[j]] % P;
			if (!(i % pr[j])) break;
			sum[i * pr[j]] = -sum[i];
		}
	}
	for(re int i = 1; i <= N; i++) sum[i] += sum[i - 1], pk[i] = (pk[i] + pk[i - 1]) % P;
}

inline int F(int n, int m)
{
	LL res = 0;
	for(re int l = 1, r; l <= min(n, m); l = r + 1)
	{
		r = min(n / (n / l), m / (m / l));
		res = (res + (LL)(sum[r] - sum[l - 1] + P) * (n / l) % P * (m / l)) % P;
	}
	return res;
}

int main()
{
	scanf("%d%d%d", &n, &m, &k);
	Euler();
	LL ans = 0;
	for(re int l = 1, r; l <= min(n, m); l = r + 1)
	{
		r = min(n / (n / l), m / (m / l));
		ans = (ans + (LL)(pk[r] - pk[l - 1] + P) * F(n / l, m / l) % P) % P;
	}
	printf("%lld
", ans);
}

但LG上有多组数据，显然太慢了

同样套路地 (T=pd)
然后这个式子成了

[sum_{T=1} lfloor frac{n}{T} floor lfloor frac{m}{T} floor sum_{d|T} d^k mu(frac{T}{d}) ]

(g(d)=d^k) 显然是个积性函数，然后 (G=g * mu) 也是个积性函数
于是我们考虑线筛预处理 (G)，然后数论分快做到单次 (O(sqrt n))

根据积性函数性质有 (G(d) = prod_{i=1} G({p_i}^{c_i}))

然后我们思考什么样的数有贡献

[G(n) = prod_{i=1} sum_{j=0}^{c_i} mu({p_i}^{j}) {p_i}^{(c_i-j)k} ]

因为 (mu) 的性质，我们知道，只有当 (j=0) 或 (j=1) 时有贡献，于是有

[egin{aligned} G(n) &= prod_{i=1} mu(1) {p_i}^{c_i k} + mu(p_i) {p_i}^{(c_i-1)k} \ &= prod_{i=1} {p_i}^{c_i k} - {p_i}^{(c_i-1)k} \ &= prod_{i=1} {p_i}^{(c_i-1) k}({p_i}^k-1) end{aligned} ]

当 (c_i = 1) 的时候，就是质数的时候，(G(p)=p^k-1)
因为 (G) 是积性函数，所以 (G(ab)=G(a)G(b)(gcd(a,b)=1))
若 (a,b) 不互质，因为在线筛时枚举质数，所以 (bin mathbb P)，设 (a = a' p^c(gcd(a,a')=1))
那么 (G(ab)=G(a')G(p^{c+1})=G(a')p^{ck}(p^k-1))
线筛过程中 (p^{(c-1)k}(p^k-1)) 已计入 (G(ab)) 中，所以本次再乘上 (p^k) 即可
综上

[G(ab)= egin{cases} G(a)G(b) & gcd(a,b)=1 \ G(a)b^k & gcd(a,b)>1 end{cases} ]

线筛即可完美处理

( ext{Code})

#include<cstdio>
#include<iostream>
#define LL long long
#define re register
using namespace std;

const int N = 5e6, P = 1e9 + 7;
int n, m, k, totp, pr[N], vis[N + 5], pk[N + 5];
LL sum[N + 5];

inline int fpow(LL x, LL y)
{
	LL res = 1;
	for(; y; y >>= 1)
	{
		if (y & 1) res = res * x % P;
		x = x * x % P;
	}
	return res;
}

inline void Euler()
{
	vis[1] = sum[1] = pk[1] = 1;
	for(re int i = 2; i <= N; i++)
	{
		if (!vis[i]) pr[++totp] = i, pk[i] = fpow(i, k), sum[i] = (pk[i] - 1 + P) % P;
		for(re int j = 1; j <= totp && i * pr[j] <= N; j++)
		{
			vis[i * pr[j]] = 1, pk[i * pr[j]] = (LL)pk[i] * pk[pr[j]] % P;
			if (i % pr[j]) sum[i * pr[j]] = sum[i] * sum[pr[j]] % P;
			else{sum[i * pr[j]] = sum[i] * pk[pr[j]] % P; break;}
		}
	}
	for(re int i = 1; i <= N; i++) sum[i] = (sum[i] + sum[i - 1]) % P;
}

int main()
{
	int T; scanf("%d%d", &T, &k);
	Euler();
	for(; T; T--)
	{
		scanf("%d%d", &n, &m);
		LL ans = 0;
		for(re int l = 1, r; l <= min(n, m); l = r + 1)
		{
			r = min(n / (n / l), m / (m / l));
			ans = (ans + (sum[r] - sum[l - 1] + P) * (n / l) % P * (m / l)) % P;
		}
		printf("%lld
", ans);
	}
}