洛谷P1447 [NOI2010]能量采 莫比乌斯反演+整除分块

---

洛谷P1447 [NOI2010]能量采集

---

标签

• 莫比乌斯反演
• 整除分块

---

前言

---

简明题意

• 给一个$n\ast m$的矩阵，如果从$(0,0)到(x,y)$的连线上点的数量是$cnt$（不包含$(0,0),(x,y)$），那么点$(x,y)$的花费就是$2\ast cnt+1$.问所有点的花费之和。

---

思路

• $(0,0)到(x,y)$连线上点的数量是$gcd(x,y)$(包含$(x,y)$)，那么按照题意得$cnt$就应该是$gcd(x,y)-1$，那么所有点得贡献和就是：
  $$\sum _{i=1}^n\sum _{j=1}^m\left(2\ast \left(gcd(i,j)-1\right)+1\right)$$
• 提出常数，原式就变成：
  $$2\ast \sum _{i=1}^n\sum _{j=1}^{m}gcd(i,j)-n\ast m$$
• 所以，现在的难点就在于如何求$\sum _{i=1}^n\sum _{j=1}^{m}gcd(i,j)$。但这个实际很简单。我们改为枚举$gcd(i,j)$，就变成：
  $$\sum _{x=1}^n\left(x\ast \sum _{i=1}^n\sum _{j=1}^m[gcd(i,j)==x]\right)$$
• 而$\sum _{i=1}^n\sum _{j=1}^m[gcd(i,j)==x]$的求法在我的另一篇博客中详细说明了，请戳这里，我们现在直接把结论摆出来：
  $$\sum _{i=1}^n\sum _{j=1}^m[gcd(i,j)==x]=\sum _{d=1}^{[\frac{n}{x}]}\left(\mu (d)\ast [\frac{n}{dx}]\ast [\frac{m}{dx}]\right)$$
• 再把这个带入原式，就得到了：
  $$\sum _{i=1}^n\sum _{j=1}^{m}gcd(i,j)=\sum _{x=1}^n\left(x\ast \sum _{d=1}^{[\frac{n}{x}]}\left(\mu (d)\ast [\frac{n}{dx}]\ast [\frac{m}{dx}]\right)\right)$$
• 预处理用线筛$O(n)$，$x$的枚举是$O(n)$，后面那一坨可以整除分块，复杂度$O(\sqrt{\frac{n}{x}})$。上面式子的复杂度是$O(n+n\ast \sqrt{\frac{n}{x}})$，就是$O(n\ast \sqrt{\frac{n}{x}})$

---

注意事项

• 无

---

总结

• $(0,0)$到$(x,y)$连线上点的数量是$gcd(x,y)$，包含$(x,y)但不包含(0,0)$

---

AC代码

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

const int maxn = 1e5 + 10;

int n, m;

bool no_prime[maxn];
int prime[maxn], mu[maxn], pre_mu[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_mu[i] = pre_mu[i - 1] + mu[i];

   return cnt;
}

int cal2(int n, int m, int x)
{
   n = n / x, m = m / x;

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


void solve()
{
   scanf("%d%d", &n, &m);
   if (m < n) swap(n, m);
   shai(m);

   long long ans = 0;
   for (int x = 1; x <= n; x++)
   	ans += 1ll * x * cal2(n, m, x);

   printf("%lld", 2 * ans - n * m);
}

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