P2522 [HAOI2011]Problem b 莫比乌斯反演+容斥+整除分块

---

P2522 [HAOI2011]Problem b

---

标签

• 莫反
• 容斥

---

前言

• 喵喵喵
• 确保你会做这一题

---

简明题意

• 给定$a,b,c,d,k$，其中$i\in [a,b]，j\in [c,d]$，现在需要你求出使得$gcd(i,j)==k$的二元组$(i,j)$的数量

---

思路

• 转换成数学形式，就是要你求：
  $$\sum _{i=a}^b\sum _{j=c}^d[gcd(i,j)==k]$$
• 现在需要确保你会计算下面的式子
  $$\sum _{i=1}^b\sum _{j=1}^d[gcd(i,j)==k]$$

上式的计算方法我在另一篇博客中详细讲解了，不清楚的同学请戳这里

• 我们假设$a=3,b=4,c=2,d=5$，那么我们枚举的就应该是：
  $$\begin{array}{clcrc}i=3& 3,2& 3,3& 3,4& 3,5\\ i=4& 4,2& 4,3& 4,4& 4,5\end{array}$$
• 而如果$a=1,b=4,c=1,d=5$，枚举的就应该是
  $$\begin{array}{clcrcc}i=1& 1,1& 1,2& 1,3& 1,4& 1,5\\ i=2& 2,1& 2,2& 2,3& 2,4& 2,5\\ i=3& 3,1& 3,2& 3,3& 3,4& 3,5\\ i=4& 4,1& 4,2& 4,3& 4,4& 4,5\end{array}$$
• 这个结果是可以直接求得（参照上一个博客），现在用不同颜色划分一下这个表格
  $$\begin{array}{clcrcc}i=1& 1,1& 1,2& 1,3& 1,4& 1,5\\ i=2& 2,1& 2,2& 2,3& 2,4& 2,5\\ i=3& 3,1& 3,2& 3,3& 3,4& 3,5\\ i=4& 4,1& 4,2& 4,3& 4,4& 4,5\end{array}$$
• 我们要求得是红色部分，而我们能求得必须是从左上角开始得，因为上一篇博客中求出了$\sum _{i=1}^n\sum _{j=1}^m[gcd(i,j)==k]$。我们发现红色部分得结果，刚好就是整个表格的结果，减去两个蓝色部分的结果。这个时候会发现紫色部分的结果减去了两次，我们就再把紫色部分的结果加一次，这样就算出正确结果了。

---

注意事项

• 无

---

总结

• 无

---

AC代码

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn = 5e4 + 10;

bool no_prime[maxn];
int prime[maxn], mu[maxn], pre[maxn];
int shai(int n)
{
   int cnt = 0;
   mu[1] = 1;

   for (int i = 2; i <= n; i++)
   {
   	if (!no_prime[i])
   		prime[++cnt] = i, mu[i] = -1;

   	for (int j = 1; j <= cnt && prime[j] * i <= n; j++)
   	{
   		no_prime[prime[j] * i] = 1;
   		mu[prime[j] * i] = i % prime[j] == 0 ? 0 : -mu[i];
   		if (i % prime[j] == 0) break;
   	}
   }

   for (int i = 1; i <= n; i++)
   	pre[i] = pre[i - 1] + mu[i];
   return cnt;
}

long long cal(int n, int m)
{
   if (n < m) swap(n, m);
   long long ans = 0;
   int l = 1, r;
   while (l <= m)
   {
   	r = min(n / (n / l), m / (m / l));
   	ans += (long long)(pre[r] - pre[l - 1]) * (n / l) * (m / l);
   	l = r + 1;
   }
   return ans;
}

void solve()
{
   shai(maxn - 10);

   int t;
   scanf("%d", &t);
   while (t--)
   {
   	int a, b, c, d, k;
   	scanf("%d%d%d%d%d", &a, &b, &c, &d, &k);

   	printf("%lld\n", cal(b / k, d / k) - cal((a - 1) / k, d / k) - cal((c - 1) / k, b / k) + cal((a - 1) / k, (c - 1) / k));
   }
}

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