codeforces-839D(莫比乌斯反演)


codeforces-839D Winter is here



图片说明



,则答案可以表示为
,设的倍数个数为,则
所以有,反演得


#include<bits/stdc++.h>
using namespace std;
const int N=1e6+6;
const int p=1e9+7;
typedef long long ll;
ll ksm(ll a,ll b){ll ans=1;for(;b>0;b>>=1,a=a*a%p)if(b&1)ans=ans*a%p;return ans;}
short int mu[N];
int flag[N],F[N],f[N];
int n;
void pre(){
    mu[1]=1;
    for(int i=1;i<N;i++)
    for(int j=i+i;j<N;j+=i)mu[j]-=mu[i];
}
int main(){
    pre();
    cin>>n;
    for(int i=1,x;i<=n;i++)scanf("%d",&x),flag[x]++;
    ll ans=0;
    for(int d=1;d<N;d++){
        int cnt=0;
        for(int j=d;j<N;j+=d){
            if(flag[j])cnt+=flag[j];
        }
        F[d]=1LL*cnt*ksm(2,cnt-1)%p;
    }
    for(int i=1;i<N;i++){
        for(int j=i;j<N;j+=i){
            f[i]+=F[j]*mu[j/i];
            if(f[i]<0)f[i]+=p;
            if(f[i]>=p)f[i]-=p;
        }
    }
    for(int d=2;d<N;d++){
        ans+=1LL*d*f[d]%p;
        ans%=p;
    }
    printf("%lld\n",ans);
}

全部评论

相关推荐

07-03 11:02
中山大学 C++
字节刚oc,但距离九月秋招很近了有两段互联网实习,非腾讯字节。不敢赌转正,现在在纠结去还是不去如果实习俩月离职会有什么后果吗
阿城我会做到的:不去后悔一辈子,能否转正取决于ld的态度,只要他不卡,答辩就是走流程,个人觉得可以冲一把
投递字节跳动等公司8个岗位
点赞 评论 收藏
分享
06-20 19:40
中原工学院 Java
网络存储:十几天不会让你拉人办卡就结束了吧?
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务