[数学+枚举]BZOJ 4921——互质序列

题目描述

现在给定一个长度为n的序列，你需要从中删除一个非空连续子序列，使得剩下至少2个数，令E为剩下数的最大公约数的期望值，S为合法的方案数，请计算E*S的值。因为这个值可能非常大，请对998244353取模输出。

解题思路

最后答案是E*S，所以跟概率没有毛线关系，答案就是所有gcd情况的加和。
因为是删除一个连续的子序列，所以剩下的数肯定是一段前缀和一段后缀的gcd。
其次我们知道前缀或后缀的gcd肯定由不超过$log_2ai$个连续片段构成，因为每次gcd至少/2。
所以我们可以提前构造前缀gcd，在构造后缀gcd的过程中求最后的答案。

#include<cstdio>
using namespace std;
const int maxn=100005,tt=998244353;
inline int _read(){
    int num=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
int f[maxn],a[maxn],now,n,lst[maxn],ans;
int gcd(int x,int y){
    if (!y) return x;else return gcd(y,x%y);
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();
    for (int i=1;i<=n;i++) a[i]=_read();
    f[1]=a[1];for (int i=2;i<=n;i++){
        f[i]=gcd(f[i-1],a[i]);
        if (f[i]==f[i-1]) lst[i]=lst[i-1];else lst[i]=i-1;
    }
    lst[0]=-1;now=0;
    for (int i=n;i>=2;i--){
        now=gcd(now,a[i]);
        for (int j=i-2;j>=0;j=lst[j]) ans=(ans+(long long)(j-lst[j])*gcd(now,f[j])%tt)%tt;
        if (i==n) ans=(ans-now+tt)%tt;
    }
    for (int i=2;i<n;i++) ans=(ans+f[i])%tt;
    printf("%d\n",ans);
    return 0;
}