Codeforces839 D. Winter is here（容斥+式子化简）

题意：

数据范围：n<=2e5，a(i)<=1e6

解法：

$$\begin{gathered} 显然只需要计算出每种gcd的所有子序列的长度和. \\ 令d[i]表示gcd为i的长度和,那么ans=\sum i\ast d[i] \\ 考虑通过容斥计算d[i]: \\ d[i]=gcd为i的倍数的长度和-d[i\ast 2]-d[i\ast 3]...... \\ 剩下的问题为gcd为i的倍数的长度和如何计算: \\ 设有cnt个i的倍数,那么长度和为: \\ C(cnt,1)\ast 1+C(cnt,2)\ast 2+...+C(cnt,cnt)\ast cnt, \\ 即\sum _{i=1}^{cnt}C(cnt,i)\ast i. \\ 试着化简一下: \\ \sum _{i=1}^{cnt}C(cnt,i)\ast i \\ =\sum _{i=1}^{cnt}\frac{cnt!}{i!\ast (cnt-i)!}\ast i \\ =\sum _{i=1}^{cnt}\frac{cnt!}{(i-1)!\ast (cnt-i)!} \\ =\sum _{i=1}^{cnt}\frac{(cnt-1)!}{(i-1)!\ast (cnt-i)!}\ast cnt \\ =\sum _{i=1}^{cnt}C(cnt-1,i-1)\ast cnt \\ =cnt\ast \sum _{i=1}^{cnt}C(cnt-1,i-1) \\ =cnt\ast \sum _{i=0}^{cnt-1}C(cnt-1,i) \\ =cnt\ast 2^{cnt-1} \end{gathered}$$

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=1e6+5;
const int mod=1e9+7;
int cnt[maxm];
int p2[maxm];
int d[maxm];
int a[maxm];
int n;
void solve(){
    p2[0]=1;
    for(int i=1;i<maxm;i++)p2[i]=p2[i-1]*2%mod;
    cin>>n;
    for(int i=1;i<=n;i++){
        int x;cin>>x;
        a[x]++;
    }
    for(int i=1;i<maxm;i++){//cnt[i]=i的倍数.
        for(int j=i;j<maxm;j+=i){
            cnt[i]+=a[j];
        }
    }
    for(int i=maxm-1;i>=2;i--){
        if(!cnt[i])continue;
        d[i]=cnt[i]*p2[cnt[i]-1]%mod;
        for(int j=i+i;j<maxm;j+=i){
            d[i]=(d[i]-d[j])%mod;
        }
    }
    int ans=0;
    for(int i=2;i<maxm;i++){
        ans=(ans+i*d[i]%mod)%mod;
    }
    ans=(ans%mod+mod)%mod;
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---