bzoj 2301: [HAOI2011]Problem b （反演）

题目描述

传送门

题目大意：
$\sum\limits_{i=a}^b \sum\limits_{j=c}^d [gcd(i,j)==k]$

题解

$\sum\limits_{i=a}^b \sum\limits_{j=c}^d [gcd(i,j)==k]$
$=\sum\limits_{i=1}^b \sum\limits_{j=1}^d [gcd(i,j)==k]-\sum\limits_{i=1}^a \sum\limits_{j=1}^d [gcd(i,j)==k]-\sum\limits_{i=1}^b \sum\limits_{j=1}^c [gcd(i,j)==k]+\sum\limits_{i=1}^a \sum\limits_{j=1}^c [gcd(i,j)==k]$

然后式子变成了类似的四部分

$$\sum\limits_{i=1}^n \sum\limits_{j=1}^m [gcd(i,j)==k]$$
$$\sum\limits_{i=1}^{\lfloor {n\over k}\rfloor}\sum\limits_{j=1}^{\lfloor {m \over k}\rfloor} [gcd(i,j)=1]$$
利用 $[n=1]=\sum\limits_{d|n} \mu(d)$
$$\sum\limits_{i=1}^{\lfloor {n\over k}\rfloor}\sum\limits_{j=1}^{\lfloor {m \over k}\rfloor} \sum_{d|(i,j)} \mu(d)$$
$$\sum_{d=1}^n \mu(d) \sum\limits_{i=1}^{\lfloor {n\over k}\rfloor}[d|i]\sum\limits_{j=1}^{\lfloor {m \over k}\rfloor} [d|j]$$
$$\sum_{d=1}^n \mu(d) {\lfloor {{n\over k}\over d}\rfloor} {\lfloor {{m \over k}\over d}\rfloor}$$
$\lfloor {n\over i} \rfloor$ 的取值最多被分成了 $\sqrt n$段相同的取值，所以我们可以将取值相同的放到一块写。
$n/i$的取值一直延续到 $n/(n/i)$
预处理前缀和，就可以用 $O(\sqrt n)$的时间求解。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define ll long long 
#define N 50000
using namespace std;
int  mu[N+3],prime[N+3],pd[N+3];
int n,m,t,a,b,c,d,k,sum[N+3];
void calc()
{
    mu[1]=1; sum[1]=mu[1];
    for (int i=2;i<=N;i++)
     {
        if (!pd[i])
         {
            prime[++prime[0]]=i;
            mu[i]=-1;
         }
        for (int j=1;j<=prime[0];j++)
        {
            if (i*prime[j]>N) break;
            pd[i*prime[j]]=1;
            if (i%prime[j]==0)
             {
                mu[i*prime[j]]=0;
                break;
             }
            else
             mu[i*prime[j]]=-mu[i];
        }
        sum[i]=sum[i-1]+mu[i];
     }
}
int solve(int a,int b)
{
    if (a>b) swap(a,b);
    int j=0; int ans=0;
    for (int i=1;i<=a;i=j+1)
    {
       j=min(a/(a/i),b/(b/i));
       ans=ans+(sum[j]-sum[i-1])*(a/i)*(b/i);
    }
    return ans;
}
int main()
{
    scanf("%d",&t);
    calc();
    for (int T=1;T<=t;T++)
     {
        scanf("%d%d%d%d%d",&a,&c,&b,&d,&k); a--; b--;
        c/=k; d/=k;  a/=k; b/=k;
        int ans=solve(c,d)-solve(a,d)-solve(b,c)+solve(a,b);
        printf("%d\n",ans);
     }
    return 0;
}