BZOJ3561 - DZY Loves Math VI

Posted

V

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了BZOJ3561 - DZY Loves Math VI相关的知识,希望对你有一定的参考价值。

Portal

Description

\(T(T\leq3)\)组测试数据。给出\(n,m(n\leq5\times10^5)\),求
\[ \sum_{i=1}^n \sum_{j=1}^m lcm(i,j)^{gcd(i,j)}\]

Solution

喜闻乐见推推推。
\[\begin{align*} ans &= \sum_{i=1}^n \sum_{j=1}^m lcm(i,j)^{gcd(i,j)} \&= \sum_{d=1}^{+∞} \sum_{i=1}^n \sum_{j=1}^m [gcd(i,j)=d](\frac{ij}{d})^d \&= \sum_{d=1}^{+∞} \sum_{i=1}^{?\frac{n}{d}?} \sum_{j=1}^{?\frac{m}{d}?} [gcd(i,j)=1](ijd)^d \&= \sum_{d=1}^{+∞} d^d \sum_{d_1=1}^{+∞}\mu(d_1) \sum_{d_1|i}^{?\frac{n}{d}?} \sum_{d_1|j}^{?\frac{m}{d}?} (ij)^d \&= \sum_{d=1}^{+∞} d^d \sum_{d_1=1}^{+∞}\mu(d_1) \sum_{i=1}^{?\frac{n}{dd_1}?} \sum_{j=1}^{?\frac{m}{dd_1}?} (id_1\cdot jd_1)^d \&= \sum_{d=1}^{+∞} d^d \sum_{d_1=1}^{+∞}\mu(d_1)d_1^{2d} \sum_{i=1}^{?\frac{n}{dd_1}?} i^d\sum_{j=1}^{?\frac{m}{dd_1}?} j^d \\end{align*}\] 于是我们枚举\(d\),预处理出\(f(x)=\sum_{i=1}^x i^d\),然后枚举\(d_1\)

根据调和级数总复杂度为\(O(nlogn)\)

Code

//DZY Loves Math VI
#include <algorithm>
#include <cstdio>
using std::swap;
typedef long long lint;
const int N=5e5+10;
const int P=1e9+7;
int prCnt,pr[N]; bool prNot[N];
int mu[N];
void init(int n)
{
    mu[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(!prNot[i]) pr[++prCnt]=i,mu[i]=-1;
        for(int j=1;j<=prCnt;j++)
        {
            int x=i*pr[j]; if(x>n) break;
            prNot[x]=true;
            if(i%pr[j]) mu[x]=-mu[i]; else break;
        }
    }
}
int powD[N],S[N];
int pow(int x,int y)
{
    lint r=1,t=x;
    for(int i=y;i;i>>=1,t=(t*t)%P) if(i&1) r=(r*t)%P;
    return r;
}
int main()
{
    int task=1; init(5e5);
    while(task--)
    {
        int n,m; scanf("%d%d",&n,&m);
        if(n>m) swap(n,m);
        int ans=0;
        for(int i=1;i<=m;i++) powD[i]=1;
        for(int d=1;d<=n;d++)
        {
            int k1=n/d,k2=m/d,ansD=0;
            for(int i=1;i<=k2;i++) powD[i]=(1LL*powD[i]*i)%P;
            for(int i=1;i<=k2;i++) S[i]=(S[i-1]+powD[i])%P;
            for(int d1=1;d1<=k1;d1++) 
            {
                lint t=1LL*(mu[d1]+P)%P*powD[d1]%P*powD[d1]%P;
                t=t*S[k1/d1]%P*S[k2/d1]%P;
                ansD=(ansD+t)%P;
            }
            ans=(ans+1LL*ansD*pow(d,d))%P;
        }
        printf("%d\n",ans);
    }
    return 0;
}

以上是关于BZOJ3561 - DZY Loves Math VI的主要内容,如果未能解决你的问题,请参考以下文章

bzoj3309DZY Loves Math

BZOJ3309: DZY Loves Math

[BZOJ3309]DZY Loves Math

Bzoj3481 DZY Loves Math III

Bzoj3309 DZY Loves Math

BZOJ3309DZY Loves Math