[莫比乌斯反演] HDU6053: [2017 多校-第2场] TrickGCD

题意

给出一个长度为n的数列A,求有多少个不同的长度为n的B数列满足下列限制:
$1 \le B_i \le A_i$
$For \ each \ pair( L , R ) (1 \le L \le R \le n), \quad gcd(B_L,B_L+1,...,B_R)≥2$
$n,a_i \le 100000$
有T组数据，$T \le 10$

题解

第一次打 $HDU$ 多校，全场抱大腿，都靠两位学长……
先转换一下，题目说的第二个条件显然就等价于 $gcd(B_1,...,B_n)>=2$.
然后可以容斥一下，就求满足 $gcd(B_1,...,B_n)=1$ 的方案数即可。
开始推式子:

$$\sum_{i_1=1}^{a_1}\sum_{i_2=1}^{a_2}...\sum_{i_n=1}^{a_n} [gcd(i_1,...,i_n)=1]$$
$$\sum_{i_1=1}^{a_1}\sum_{i_2=1}^{a_2}...\sum_{i_n=1}^{a_n} \sum_{d|gcd(..)} \mu(k)$$
$$\sum_{d=1}^{min(a)} \sum_{d|i_1}^{a_1}\sum_{d|i_2}^{a_2}...\sum_{d|i_n}^{a_n} \mu(d)$$
$$\sum_{d=1}^{min(a)}\mu(d) \prod_{i=1}^n \lfloor \frac {a_i} d\rfloor$$
然后如果按照传统方法枚举除法分块搞， $\prod_{i=1}^n \lfloor \frac {a_i} d\rfloor$ 有 $n\sqrt n$ 段。这样是 $O(n^2\sqrt n*T)$ 的，会 $T$ 掉。
怎么做呢？可以换个思路，对于每次算 $\prod$，我们枚举 $\lfloor \frac {a_i} d\rfloor$ 的值，然后统计每个值有几个，快速幂即可。
具体来说：
$$\sum_{d=1}^{min(a)}\mu(d) \prod_{t=1}^{max(a)/d} t^x,\quad (x 为满足 \lfloor \frac {a_i} d\rfloor=t 的 i 的个数)$$
$x$怎么求？ $\lfloor \frac {a_i} d\rfloor=t \Leftrightarrow dt\le a_i \le d(t+1)-1$，前缀和就好啦。
复杂度的话 $n+n/2+n/3+...$ 这样的东西约为 $n \log n$ 。在加上快速幂，共 $O(n \log^2 n*T)$。
看上去好像有点危险，但是快速幂的 $log$ 很小，常数也小，是可以过的。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline char gc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int getint(){
    char ch=gc(); int res=0,ff=1;
    while(!('0'<=ch&&ch<='9')) ch=='-'?ff=-1:0, ch=gc();
    while('0'<=ch&&ch<='9') res=(res<<3)+(res<<1)+ch-'0', ch=gc();
    return res*ff;
}
const int maxn=200005, N=200000, MOD=1000000007;
typedef long long LL;
int _test,n,ans,_min,_max,a[maxn],sum[maxn],p[maxn],mu[maxn];
bool vis[maxn];
void get_mu(){
    memset(vis,1,sizeof(vis));
    mu[1]=1;
    for(int i=2;i<=N;i++){
        if(vis[i]) p[++p[0]]=i, mu[i]=-1;
        for(int j=1;j<=p[0]&&(LL)i*p[j]<=N;j++){
            vis[i*p[j]]=false;
            if(i%p[j]==0){ mu[i*p[j]]=0; break; }
            mu[i*p[j]]=-mu[i];
        }
    }
}
int Pow(int a,int b){
    LL res=1;
    for(LL w=a;b;b>>=1,w=w*w%MOD) if(b&1) res=(res*w)%MOD;
    return res;
}
void Solve(){
    ans=0;
    for(int d=1;d<=_min;d++){
        LL res=(mu[d]+MOD)%MOD;
        for(int t=1;t<=_max/d;t++) res=res*Pow(t,sum[(t+1)*d-1]-sum[t*d-1])%MOD;
        ans=(ans+res)%MOD;
    }
}
int main(){
    freopen("hdu6053.in","r",stdin);
    freopen("hdu6053.out","w",stdout);
    get_mu();
    scanf("%d",&_test);
    for(int ii=1;ii<=_test;ii++){
        memset(sum,0,sizeof(sum));
        n=getint(); _min=1e+9; _max=0; LL tot=1;
        for(int i=1;i<=n;i++) sum[a[i]=getint()]++, tot=(tot*a[i])%MOD, _min=min(_min,a[i]), _max=max(_max,a[i]);
        for(int i=1;i<=N;i++) sum[i]+=sum[i-1];
        Solve();
        printf("Case #%d: %d\n",ii,((tot-ans)%MOD+MOD)%MOD);
    }
    return 0;
}