bzoj3930 [CQOI2015]选数(容斥+递推+数论)

首先我们有:在[l,R]中选若干不完全相同的数,他们的gcd<=R-L。反证即可。
我们设f[i]表示gcd恰好为K*i的个数。(选若干不完全相同的数),考虑容斥计算,倒着推即可。
f[i]=(R-L)^n-(R-L)-f[i*j] (R=r/(ki),L=(l-1)/(ki))
最后特判一下是否可以是完全相同的K。
复杂度 O(nlogn)

#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
#define mod 1000000007
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int f[N];//f[i]--gcd==i*k的个数
inline int ksm(int x,int k){
    int res=1;for(;k;k>>=1,x=(ll)x*x%mod) if(k&1) res=(ll)res*x%mod;return res;
}
int main(){
//  freopen("a.in","r",stdin);
    int n=read(),K=read(),L=read(),R=read(),lim=(R-L)/K;
    for(int i=lim;i>=1;--i){
        int l=(L-1)/(K*i),r=R/(K*i);
        f[i]=ksm(r-l,n)-(r-l);f[i]%=mod;
        for(int j=2;i*j<=lim;++j) f[i]-=f[i*j],f[i]%=mod;
        if(f[i]<0) f[i]+=mod;
    }if(L<=K&&K<=R) f[1]++;
    printf("%d\n",f[1]%mod);
    return 0;
}

你可能感兴趣的:(bzoj,容斥原理,gcd,递推)