[洛谷][POI2007]ZAP-Queries-數論

水到了一道莫比烏斯反演的模板題，凱森QvQ

【題目地址】

---

題意簡述

多組詢問，每次給定$n,m,d$，求下面式子的值：

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

---

直接莫比烏斯反演，線性求$\mu$就好啦，式子就變成下面的樣子：

$\sum_{d=1}^n\mu(d)\left\lfloor\frac{n}{d}\right\rfloor\left\lfloor\frac{m}{d}\right\rfloor$

分塊求，每次$O(\sqrt n)$就可以了。

---

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int M=1e5+10,MAX=50000;
ll prime[M],cnt;
ll mu[M];
bool vis[M];
ll n,m,d;
void init(){
	mu[1]=1;
	for(int i=2;i<=MAX;i++){
		if(!vis[i]){
			prime[++cnt]=i;
			mu[i]=-1;
		}
		for(int j=1,v;j<=cnt&&i*prime[j]<=MAX;j++){
			v=i*prime[j];
			vis[v]=1;
			if(!(i%prime[j])){
				break;
			}
			mu[v]=-mu[i];
		}
	}
	for(int i=1;i<=MAX;i++)mu[i]+=mu[i-1];
}
ll solve(){
	if(n>m)swap(n,m);
	ll a=n/d,b=m/d,ans=0;
	for(ll i=1,j;i<=a;i=j+1){
		j=min(a/(a/i),b/(b/i));
		ans+=((mu[j]-mu[i-1])*(a/i)*(b/i));
	}
	return ans;
}
int T;
int main(){
	init();
	for(scanf("%d",&T);T--;){
		scanf("%lld%lld%lld",&n,&m,&d);
		printf("%lld\n",solve());
	}
	return 0;
}