[HAOI2011]Problem b


原題鏈接

題目描述

  • 對於給出的n個詢問,每次求有多少個數對(x,y),滿足a≤x≤b,c≤y≤d,且gcd(x,y) = k,gcd(x,y)函數為x和y的最大公約數。

輸入輸出格式

  • 輸入格式:
    第一行一個整數n,接下來n行每行五個整數,分別表示a、b、c、d、k
  • 輸出格式:
    共n行,每行一個整數表示滿足要求的數對(x,y)的個數

解題思路

  • 這個題要求的其實就是:\(Ans=\sum_{i=a}^{b}\sum_{j=c}^{d}[gcd(i,j)=k]\)
  • 如果做過一道叫:[POI2007]ZAP-Queries的題,那么這題就顯得非常的簡單了。因為那道題就是這道題的一個特殊情況\((a=1,c=1)\)
  • 我們可以發現本題所算的\(a\)~\(b\),\(c\)~\(d\)的答案,實質上由一個簡單的容斥就可以轉換成\(Ans((1,b),(1,d))-Ans((1,b),(1,c-1))-Ans((1,a-1),(1,d))+Ans((1,a-1),(1,c-1))\),也就是一種類似於前綴和的容斥。具體的原因,其實把\(\sum\)隨便手寫幾項,就可以發現這一定是正確的。
  • 至於如何求\(1\)~\(n\),\(1\)~\(m\),就按照那道題化簡一下式子就可以了。
  • 我們設:

\[f(k)=\sum_{i=1}^{a}\sum_{j=1}^{b}[gcd(i,j)=k] \]

\[F(n)=\sum_{n|k}f(k)=\lfloor\frac{a}{n}\rfloor\lfloor\frac{b}{n}\rfloor \]

則可以由莫比烏斯反演可以推出:

\[f(n)=\sum_{n|k}\mu(\lfloor\frac{k}{n}\rfloor)F(k) \]

  • (PS:如果不知道為什么要設這兩個函數,可以點開我上面放的鏈接)
  • 設完這兩個函數之后,我們便驚喜的發現,\(Ans=f(k)\)
  • 於是就直接開始推答案:

\[Ans=\sum_{k|n}\mu(\lfloor\frac{n}{k}\rfloor)F(n) \]

枚舉\(\lfloor\frac{n}{k}\rfloor\)設為\(t\)

\[Ans=\sum_{t=1}^{min(\lfloor\frac{a}{k}\rfloor,\lfloor\frac{b}{k}\rfloor)}\mu(t)\lfloor\frac{a}{tk}\rfloor\lfloor\frac{b}{tk}\rfloor \]

這時候,這個式子已經可以做到\(O(n)\)的時間復雜度了,但是因為有多組數據,所以我們再用一下整除分塊,這式子就可以做到\(O(\sqrt{n})\)了。

  • 我們只需要寫一個這樣的函數,每次詢問調用四遍就可以了。

下附代碼:

// luogu-judger-enable-o2
#include<bits/stdc++.h>
#define N 60010
using namespace std;
inline void read(int &x)
{
    x=0;
    static int p;p=1;
    static char c;c=getchar();
    while(!isdigit(c)){if(c=='-')p=-1;c=getchar();}
    while(isdigit(c)) {x=(x<<1)+(x<<3)+(c-48);c=getchar();}
    x*=p;	
}
bool vis[N];
int prim[N],mu[N],sum[N],cnt,k;
void get_mu(int n)
{
    mu[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(!vis[i]){mu[i]=-1;prim[++cnt]=i;}
        for(int j=1;j<=cnt&&i*prim[j]<=n;j++)
        {
            vis[i*prim[j]]=1;
            if(i%prim[j]==0)break;
            else mu[i*prim[j]]=-mu[i];
        }
    }
    for(int i=1;i<=n;i++)sum[i]=sum[i-1]+mu[i];
}
long long calc(int a,int b)
{
    static int max_rep;
    static long long ans;
    max_rep=min(a,b);ans=0;
    for(int l=1,r;l<=max_rep;l=r+1)
    {
        r=min(a/(a/l),b/(b/l));
        ans+=(1ll*a/(1ll*l*k))*(1ll*b/(1ll*l*k))*(sum[r]-sum[l-1]);
    }
    return ans;
}
int main()
{
//	freopen("P3455.in","r",stdin);
//	freopen("P3455.out","w",stdout);
    int t;
    read(t);
    get_mu(50000);
    while(t--)
    {
        static int a,b,c,d;
        read(a);read(b);read(c);read(d);read(k);
        printf("%lld\n",calc(b,d)-calc(b,c-1)-calc(a-1,d)+calc(a-1,c-1));
    }
    return 0;
}


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM