【BZOJ2820】GCD

最新推荐文章于 2024-09-02 13:47:18 发布

ezoixx174

最新推荐文章于 2024-09-02 13:47:18 发布

阅读量7.9k

点赞数 2

分类专栏：莫比乌斯反演

本文链接：https://blog.csdn.net/ezoixx174/article/details/81572022

版权

莫比乌斯反演专栏收录该内容

1 篇文章 0 订阅

订阅专栏

博客介绍了BZOJ2820题目的解决方案，利用莫比乌斯反演进行求解，并通过设置辅助变量dp优化时间复杂度，以应对多组数据的挑战。最后提供了AC代码，总时间复杂度为O(T⋅(π(n)logπ(n)+n))。

摘要由CSDN通过智能技术生成

题目

链接：BZOJ2820
（权限题，洛谷P2257）

解法：莫比乌斯反演

一道较简单的莫比乌斯反演，推推式子就OK了。

A n s = \sum i = 1 n \sum j = 1 m \sum p [gcd (i, j) = p] = \sum p \sum i ⌊ n p ⌋ \sum j ⌊ m p ⌋ [gcd (i, j) = 1] = \sum p \sum i ⌊ n p ⌋ \sum j ⌊ m p ⌋ \sum d | gcd (i, j) μ (d) = \sum p \sum d = 1 min (⌊ n p ⌋, ⌊ m p ⌋) μ (d) \sum i ⌊ n d p ⌋ \sum j ⌊ m d p ⌋ 1 = \sum p \sum d = 1 min (⌊ n p ⌋, ⌊ m p ⌋) ⌊ n d p ⌋ ⌊ m d p ⌋ μ (d)

$\begin{split} Ans&=\sum_{i=1}^n\sum_{j=1}^m\sum_p\left[\gcd\left(i,j\right)=p\right]\\ &=\sum_p\sum_i^{\left\lfloor\frac{n}{p}\right\rfloor}\sum_j^{\left\lfloor\frac{m}{p}\right\rfloor}\left[\gcd\left(i,j\right)=1\right]\\ &=\sum_p\sum_i^{\left\lfloor\frac{n}{p}\right\rfloor}\sum_j^{\left\lfloor\frac{m}{p}\right\rfloor}\sum_{d|\gcd\left(i,j\right)}\mu(d)\\ &=\sum_p\sum_{d=1}^{\min\left(\left\lfloor\frac{n}{p}\right\rfloor,\left\lfloor\frac{m}{p}\right\rfloor\right)}\mu(d)\sum_i^{\left\lfloor\frac{n}{dp}\right\rfloor}\sum_j^{\left\lfloor\frac{m}{dp}\right\rfloor}1\\ &=\sum_p\sum_{d=1}^{\min\left(\left\lfloor\frac{n}{p}\right\rfloor,\left\lfloor\frac{m}{p}\right\rfloor\right)}\left\lfloor\frac{n}{dp}\right\rfloor\left\lfloor\frac{m}{dp}\right\rfloor\mu\left(d\right) \end{split}$

单次询问时间复杂度： $O\left(\pi(n)\sqrt{\min\left(n,m\right)}\right)$ ，一眼看过去，诶，能过，突然就看到了多组数据 $T\leqslant10^5$ 。。凉了呀

发现式中有迭代变量的乘积，于是设 $t=dp$ ，就有：

A n s = \sum p \sum d = 1 min (⌊ n p ⌋, ⌊ m p ⌋) ⌊ n d p ⌋ ⌊ m d p ⌋ μ (d) = \sum t = d p ⌊ n t ⌋ ⌊ m t ⌋ \sum p μ (t p)

$\begin{split} Ans&=\sum_p\sum_{d=1}^{\min\left(\left\lfloor\frac{n} {p}\right\rfloor,\left\lfloor\frac{m} {p}\right\rfloor\right)}\left\lfloor\frac{n}{dp}\right\rfloor\left\lfloor\frac{m}{dp}\right\rfloor\mu\left(d\right)\\ &=\sum_{t=dp}\left\lfloor\frac{n}{t}\right\rfloor\left\lfloor\frac{m}{t}\right\rfloor\sum_p\mu\left(\frac{t}{p}\right) \end{split}$

在筛出素数后，迭代求出 $\sum_p\mu\left(\frac{t}{p}\right)$ 的前缀和，
即可使用分块求解。

时间复杂度： $O\left(T\cdot(\pi(n)\log\pi(n)+\sqrt{n}\right))$

AC代码

#include<iostream>
#include<cstdio>
#include<vector>

using namespace std;

const int N=10000000;
vector<int> prime;
bool check[10000001];
int T,n,m;
long long summu[10000001],mu[10000001];

void sieve(){
    mu[1]=1ll;
    for(int i=2;i<=N;++i){
        if(!check[i])prime.push_back(i),mu[i]=-1ll;
        for(int j:prime){
            if(i*j>N)break;
            mu[i*j]=-mu[i],check[i*j]=true;
            if(i%j==0){mu[i*j]=0;break;}
        }
    }
    for(int i:prime)for(int j=1;i*j<=N;++j)summu[i*j]+=mu[j];
    for(int i=1;i<=N;++i)summu[i]+=summu[i-1];
}

void solve(int n,int m){
    if(n>m)swap(n,m);
    int lst;long long res=0ll;
    for(int i=1;i<=n;i=lst+1)lst=min(n/(n/i),m/(m/i)),res+=(summu[lst]-summu[i-1])*(n/i)*(m/i);
    printf("%lld\n",res);
}

int main(){
    sieve();
    scanf("%d",&T);for(int i=1;i<=T;++i)scanf("%d%d",&n,&m),solve(n,m);
}