码迷,mamicode.com
首页 > 其他好文 > 详细

简单的数学题

时间:2019-10-04 20:41:21      阅读:118      评论:0      收藏:0      [点我收藏+]

标签:等于   ref   此刻   blog   inline   cassert   getchar   tchar   预处理   

题目:link


原式
\[\sum_{i=1}^{n}\sum_{j=1}^n{ijgcd(i,j)}\]
枚举 \(gcd\) 得到
\[\sum_{d=1}^n\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum_{j=1}^{\lfloor\frac{n}{d}\rfloor}ijd^3[gcd(i,j)=1]\]
\(d^3\) 提到前面
\[\sum_{d=1}^nd^3\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum_{j=1}^{\lfloor\frac{n}{d}\rfloor}ij[gcd(i,j)=1]\]
设函数
\[f(n)=\sum_{i=1}^{n}\sum_{j=1}^nij[gcd(i,j)=1]\]
\[s(n)=\sum_{i=1}^{n}\sum_{j=1}^nij\]
则原式等于
\[\sum_{d=1}^nd^3f(\lfloor\frac{n}{d}\rfloor)\]
可以看出是数论分块,其中立方部分和为
\[\sum_{i=1}^{n}i^3=(\sum_{i=1}^ni)^2=[\frac{n(n+1)}{2}]^2\]
求出前缀做差即可


考虑快速求 \(f(n)\),非常显然
\[s(n)=\sum_{i=1}^{n}\sum_{j=1}^nij=[\frac{n(n+1)}{2}]^2\]
而对于 \(s(n)\) 同样枚举 \(gcd\) 可得
\[s(n)=\sum_{i=1}^{n}\sum_{j=1}^nij=\sum_{t=1}^{n}\sum_{i=1}^{\lfloor\frac{n}{t}\rfloor}\sum_{j=1}^{\lfloor\frac{n}{t}\rfloor}t^2ij[gcd(i,j)=1]\]
\(f(n)\) 替换上式得
\[s(n)=\sum_{t=1}^{n}t^2f(\lfloor\frac{n}{t}\rfloor)=f(n)+\sum_{t>1}t^2f(\lfloor\frac{n}{t}\rfloor)\]
所以有
\[f(n)=s(n)-\sum_{t>1}t^2f(\lfloor\frac{n}{t}\rfloor)=[\frac{n(n+1)}{2}]^2-\sum_{t>1}t^2f(\lfloor\frac{n}{t}\rfloor)\]
到了此刻已经非常容易求了,数论分块即可,其中
\[\sum_{i=1}^{n}i^2=\frac{n(n+1)(2n+1)}{6}\]
做法同立方和,前缀做差


考虑预处理 \(f(n)\) 部分
\[f(n)=\sum_{i=1}^{n}\sum_{j=1}^nij[gcd(i,j)=1]=f(n-1)+2\sum_{i=1}^{n}in[gcd(i,n)=1]\]

那么需要快速求得
\[\sum_{i=1}^ni[gcd(i,n)=1]\]
发现这就是与 \(n\) 互质的数的和,就等于
\[n\frac{\varphi(n)}{2}\]
详细证明参考链接

带回原式得到
\[f(n)=f(n-1)+n^2\varphi(n)\]
线性筛预处理即可


Code:

#pragma GCC optimize(2)
#include<stdio.h>
#include<string.h>
#include<cassert>
#define Mod p
#define LL long long
#define N 5000000

template<class T>
inline void read(T &x){
    x=0;char c=getchar();T flag=1;
    while(c<'0'||c>'9'){if(c=='-')flag=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+c-48;c=getchar();}
    x*=flag;
}

LL n,p,inv,inv6;
int cnt,P[N];
bool mark[N];
LL phi[N],f[N],F[N];

inline LL qpow(LL a,LL b){
    if(!b) return 1LL;
    if(b==1LL) return a;
    LL t=qpow(a,b>>1LL);
    t=t*t%Mod;
    if(b&1LL) t=t*a%Mod;
    assert(t>=0);
    return t;
}
inline LL pow2(LL x){//平方前缀和
    return ((inv6*x)%Mod)*(((x+1LL)%Mod)*((2LL*x+1LL)%Mod)%Mod)%Mod;
}
inline LL pow2(LL l,LL r){平方区间和
    return ((pow2(r)-pow2(l-1LL))%Mod+Mod)%Mod;
}
inline LL pow3(LL x){立方前缀和
    return qpow(((x%Mod)*(x+1LL)%Mod)*inv%Mod,2LL);
}
inline LL pow3(LL l,LL r){立方区间和
    return ((pow3(r)-pow3(l-1LL))%Mod+Mod)%Mod;
}
inline void Prework(int x){//线性筛预处理 f(n)
    cnt=0;
    f[1]=phi[1]=1;
    for(int i=2;i<=x;++i){
        if(!mark[i]){
            P[cnt++]=i;
            phi[i]=i-1;
        }
        for(int j=0;j<cnt&&i*P[j]<=x;++j){
            mark[i*P[j]]=1;
            if(!(i%P[j])){
                phi[i*P[j]]=phi[i]*P[j];
                break;
            }else phi[i*P[j]]=phi[i]*(P[j]-1);
        }
        LL ret=qpow(i,2LL)*phi[i]%Mod;
        f[i]=(f[i-1]+ret)%Mod;
    }
}
inline LL solve(LL x){//杜教筛
    if(x<=N-3) return f[x];
    if(~F[n/x]) return F[n/x];
    F[n/x]=qpow(((x%Mod)*((x+1)%Mod)%Mod)*inv%Mod,2);
    for(LL l=2,r;l<=x;l=r+1){
        r=x/(x/l);
        F[n/x]=((F[n/x]-(pow2(l,r)*solve(x/l)%Mod))%Mod+Mod)%Mod;
    }
    return F[n/x];
}
int main(){
    memset(F,-1,sizeof(F));
    read(p),read(n);
    inv=qpow(2LL,p-2);//2的逆元
    inv6=qpow(6LL,p-2);//6的逆元
    Prework(N-3);
    LL ans=0;
    for(LL l=1,r;l<=n;l=r+1){//原式
        r=n/(n/l);
        ans=(ans+(pow3(l,r)*solve(n/l))%Mod)%Mod;
    }
    printf("%lld",ans);
}
/*
998244353 2
*/

简单的数学题

标签:等于   ref   此刻   blog   inline   cassert   getchar   tchar   预处理   

原文地址:https://www.cnblogs.com/wwlwQWQ/p/11622943.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!