bzoj3930 [CQOI2015]选数(反演||容斥原理)

题目链接

分析:
自己YY的解法:
gcd=k g c d = k ,那么我们就可以把选出的数都分解成 (ka0,ka1,...,kan) ( k ∗ a 0 , k ∗ a 1 , . . . , k ∗ a n )
那么 gcd(a0,a1,...,an)=1 g c d ( a 0 , a 1 , . . . , a n ) = 1 ,其中一定能找到两个数满足 gcd(ai,aj)=1 g c d ( a i , a j ) = 1 ,其余数字可以任意选择

[lk,rk] [ l ∗ k , r ∗ k ] 包含在 [L,R] [ L , R ] 内且 [l,r] [ l , r ] 尽量大,那么 ai a i 的范围就是 [l,r] [ l , r ]

ans=(i=lrj=lr[gcd(i,j)=1]1)(rl+1)k2 a n s = ( ∑ i = l r ∑ j = l r [ g c d ( i , j ) = 1 ] − 1 ) ∗ ( r − l + 1 ) k − 2

对于 ni=1nj=1[gcd(i,j)=1]=2ni=1ϕ(i) ∑ i = 1 n ∑ j = 1 n [ g c d ( i , j ) = 1 ] = 2 ∗ ∑ i = 1 n ϕ ( i )

S(n)=ni=1ϕ(i)=ni=1(id|i,d<iϕ(d))=n(n+1)2ni=2S(ni) S ( n ) = ∑ i = 1 n ϕ ( i ) = ∑ i = 1 n ( i − ∑ d | i , d < i ϕ ( d ) ) = n ( n + 1 ) 2 − ∑ i = 2 n S ( n i )

然而T掉了。。。


反演正解:

%%%PoPoQQQ
bzoj3930 [CQOI2015]选数(反演||容斥原理)_第1张图片

我化出来的式子是这样的:
ni=1μ(i)=S(n)=1ni=2μ(ni) ∑ i = 1 n μ ( i ) = S ( n ) = 1 − ∑ i = 2 n μ ( n i )
用这个式子计算μ的前缀和也是可以的

tip

一开始还是T,时间卡的太紧

if(n/i-1) ans-=(Sum(last)-Sum(i-1))*(n/i-1);

加了一个小特判,时间就减少了一多半:4000ms

#include
#include
#include
#include
#define ll long long

using namespace std;

const int INF=1e9+7;
const int N=1e7+5;
const ll p=1e9+7;
ll n,m,L,R;
int mu[N],sshu[N],tot=0;
bool no[N];
map<int,ll> mp;

void prepare() {
    mu[1]=1;
    for (int i=2;iif (!no[i]) {
            sshu[++tot]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=tot&&sshu[j]*ino[sshu[j]*i]=1;
            if (i%sshu[j]==0) {
                mu[i*sshu[j]]=0;
                break;
            }
            mu[i*sshu[j]]=-mu[i];
        }
    }
    for (int i=2;i1]+mu[i];
}

ll KSM(ll a,ll b) {
    ll t=1;
    while (b) {
        if (b&1) t=(t*a)%p;
        b>>=1;
        a=(a*a)%p;
    }
    return t%p;
}

ll Sum(int n) {
    if (nreturn mu[n];
    if (mp.find(n)!=mp.end()) return mp[n];
    ll ans=1,last;
    for (ll i=1;i<=n;i=last+1) {
        last=n/(n/i);
        if(n/i-1) ans-=(Sum(last)-Sum(i-1))*(n/i-1);
    }
    mp[n]=ans;
    return ans;
}

ll solve() {
    ll last,ans=0;
    for (ll i=1;i<=R;i=last+1) {
        last=min(R/(R/i),L/i?(L/(L/i)):INF);
        ans+=(Sum(last)-Sum(i-1))*KSM(R/i-L/i,n);
        ans%=p;
    }
    return (ans%p+p)%p;
}

int main()
{
    scanf("%lld%lld%lld%lld",&n,&m,&L,&R);
    L=(L-1)/m; R=R/m;
    prepare();
    printf("%lld",solve());
}

当然dada们还有其他解法

容斥:

YJQQQAQ
我们用 f[i] f [ i ] 表示 gcd g c d 刚好为 ki k ∗ i 的选数方案,那么 f[1] f [ 1 ] 就是答案,我们就for i = 100000 to 1,从大往小算
对于每一个 i i ,先算在这个区间里面有多少个 ki k ∗ i 的倍数,然后 N N 次方之
然后减去 f[ij](j=1,2,3.MAXJ) f [ i ∗ j ] ( j = 1 , 2 , 3 … . M A X J )
这就是 gcd g c d ki k ∗ i 的方案数

ij i ∗ j 100000 100000 就好了,因为这个区间最多只有 100000 100000
可能 ij i ∗ j 100000 100000 以上还会有需要减掉的,我们考虑计算这些数对答案的贡献
首先同一个 gcd g c d 只会有1个数,这些 gcd g c d 的个数就是在原区间里面除以 K>100000 K > 100000 且为 i i 的倍数的数

你可能感兴趣的:(反演)