bzoj3930 [CQOI2015]选数（反演||容斥原理）

题目链接

分析：
自己YY的解法：
$gcd=k$，那么我们就可以把选出的数都分解成$(k*a_0,k*a_1,...,k*a_n)$
那么$gcd(a_0,a_1,...,a_n)=1$，其中一定能找到两个数满足$gcd(a_i,a_j)=1$，其余数字可以任意选择

设$[l*k,r*k]$包含在$[L,R]$内且$[l,r]$尽量大，那么$a_i$的范围就是$[l,r]$

$$ans=(\sum_{i=l}^{r}\sum_{j=l}^{r}[gcd(i,j)=1]-1)*(r-l+1)^{k-2}$$

对于$\sum_{i=1}^n\sum_{j=1}^n[gcd(i,j)=1]=2*\sum_{i=1}^n\phi(i)$

$S(n)=\sum_{i=1}^n\phi(i)=\sum_{i=1}^{n}(i-\sum_{d|i,d<i}\phi(d))={n(n+1) \over 2}-\sum_{i=2}^{n}S({n \over i})$

然而T掉了。。。

---

反演正解：

%%%PoPoQQQ

我化出来的式子是这样的：
$\sum_{i=1}^{n}μ(i)=S(n)=1-\sum_{i=2}^{n}μ({n \over i})$
用这个式子计算μ的前缀和也是可以的

tip

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

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

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

#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#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;i<N;i++) {
        if (!no[i]) {
            sshu[++tot]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=tot&&sshu[j]*i<N;j++) {
            no[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;i<N;i++) mu[i]=mu[i-1]+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 (n<N) return 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]$表示$gcd$刚好为$k*i$的选数方案，那么$f[1]$就是答案，我们就for i = 100000 to 1，从大往小算
对于每一个$i$，先算在这个区间里面有多少个$k*i$的倍数，然后$N$次方之
然后减去$f[i*j](j = 1,2,3….MAXJ)$
这就是$gcd$为$k*i$的方案数

$i*j$到$100000$就好了，因为这个区间最多只有$100000$长
可能$i*j$在$100000$以上还会有需要减掉的，我们考虑计算这些数对答案的贡献
首先同一个$gcd$只会有1个数，这些$gcd$的个数就是在原区间里面除以$K>100000$且为 i i <script type="math/tex" id="MathJax-Element-64">i</script>的倍数的数