CF645F Cowslip Collections

一、题目

点此看题

题目翻译

有一个长度为$n$的数组，每次向其插入一个数，问从中选$k$个数求$\gcd$的和，输出插入数字后的值。

二、解法

看到$\gcd$就可以往莫比乌斯反演的方向想了。

设$f(i)$为$\gcd$为$i$的方案数，答案为：$\sum i\times f(i)$

设$F(i)$为$\gcd$是$i$的倍数的方案数，设$cnt[i]$数组中为$i$倍数的个数

$$F(i)=\sum _{i|d}f(d)=C(cnt[i],k)$$$$f(i)=\sum _{i|d}F(d)\times \mu (\frac{d}{i})$$考虑加入一个数的答案的变化，首先直接影响到了$cnt$数组，设$x$为这个数的因数，那么$cnt[x]$加$1$，结合上面的式子，不难发现答案会增加：
$$(C(cnt[x],k)-C(cnt[x]-1,k))\sum _{i|x}i\times \mu (i)$$我们把一开始的数组当前空，然后插入$n+q$个数，明显后面的式子可以$O(n\log n)$预处理，那么总的复杂度就是$O(n\sqrt{n})$，贴个代码$qwq$

#include <cstdio>
#include <iostream>
using namespace std;
#define int long long
const int M = 1000005;
const int jzm = 1e9+7;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int n,k,q,ans,cnt,p[M],f[M],vis[M],mu[M],g[M],inv[M],fac[M];
void init(int n)
{
    inv[0]=inv[1]=fac[0]=fac[1]=1;
    for(int i=2;i<=n;i++) inv[i]=inv[jzm%i]*(jzm-jzm/i)%jzm;
    for(int i=2;i<=n;i++) fac[i]=fac[i-1]*i%jzm;
    for(int i=2;i<=n;i++) inv[i]=inv[i]*inv[i-1]%jzm;
    mu[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(!vis[i])
        {
            p[++cnt]=i;
            mu[i]=-1;
        }
        for(int j=1;j<=cnt && i*p[j]<=n;j++)
        {
            vis[i*p[j]]=1;
            if(i%p[j]==0) break;
            mu[i*p[j]]=-mu[i];
        }
    }
    for(int i=1;i<=n;i++)
        for(int j=i;j<=n;j+=i)
            g[j]=(g[j]+i*mu[j/i])%jzm;
}
int C(int n,int m)
{
    if(n<m) return 0;
    return fac[n]*inv[m]%jzm*inv[n-m]%jzm;
}
void upd(int x)
{
    f[x]++;
    ans=(ans+(C(f[x],k)-C(f[x]-1,k))%jzm*g[x]%jzm)%jzm;
}
void dec(int x)
{
    for(int i=1;i*i<=x;i++)
        if(x%i==0)
        {
            upd(i);
            if(i*i!=x) upd(x/i);
        }
}
signed main()
{
    n=read();k=read();q=read();
    init(1e6);
    for(int i=1;i<=n;i++)
        dec(read());
    for(int i=1;i<=q;i++)
    {
        dec(read());
        printf("%lld\n",(ans+jzm)%jzm);
    }
}