P2257 莫比乌斯反演

题目传送门

题意：

计算  。

数据范围： 。

题解：

算到这一步，需要换元降低时间复杂度。令  。

可以  筛出来，具体操作就是枚举素数的倍数。

这样就可以前缀和 + 整除分块求  了。

感受：

又被卡常，把能用int的地方全用int就好了。

学会了一种优化方法：自己筛一个东西。

代码：

#include<bits/stdc++.h>
using namespace std ;
typedef long long ll ;
const int maxn = 1e7 + 5 ;
bool vis[maxn] ;
int prime[maxn] ;
int mu[maxn] , pre[maxn] ;
int f[maxn] , g[maxn] ;
int cnt = 0 ;
void get_mu(int n)
{
   mu[1] = 1 ;
   pre[1] = 1 ;
   for(int i = 2 ; i <= n ; i ++)
   {
     if(!vis[i])  prime[++ cnt] = i , mu[i] = -1 ;
     for(int j = 1 ; j <= cnt && prime[j] * i <= n ; j ++)
     {
       vis[prime[j] * i] = 1 ;
       if(i % prime[j] == 0)  break ;
        else  mu[i * prime[j]] = -mu[i] ;
     }
     pre[i] = pre[i - 1] + mu[i] ;
   }
}
void init(int n)
{
	for(int i = 1 ; i <= cnt ; i ++)
	  for(int j = prime[i] ; j <= n ; j += prime[i])
	    f[j] += mu[j / prime[i]] ;
	for(int i = 1 ; i <= n ; i ++)  g[i] = g[i - 1] + f[i] ;
}
ll solve(int n , int m)
{
   ll ans = 0 ;
   for(int l = 1 , r ; l <= n ; l = r + 1)
   {
     r = min(n / (n / l) , m / (m / l)) ;
     ans += ll(g[r] - g[l - 1]) * (n / l) * (m / l) ;
   }
   return ans ;
}
int main()
{
	int num = 1e7 ;
	get_mu(num) ;
	init(num) ;
	int t ;
	scanf("%d" , &t) ;
	while(t --)
	{
		int n , m ;
		scanf("%d%d" , &n , &m) ;
		if(n > m)  swap(n , m) ;
		printf("%lld\n" , solve(n , m)) ;
	}
	return 0 ;
}