题目大意
随机序列a。
a0=0。
ai以pi%为ai-1再加1,否则为0。
求序列和的平方的期望。
DP
设fi表示1~i的和的平方期望,gi则表示和的期望。
根据(a+b)^2=a^2+b^2+2ab
fi=∑i−1j=−1(fj+s2[i−j−1]+2∗gj∗s1[i−j−1])∗(1−pj+1)∗Πi−1k=j+2pk
g也差不多自己推推吧……
相当于枚举上一个0的位置,s1和s2表示自然数和或自然数和的平方。
这样我们可以n^2啦。
接下来把各种东西拆出来,大概维护10个值就可以线性。
这题比较麻烦但不难,自己推吧……
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=2000000+10,mo=998244353;
int f[maxn],g[maxn],p[maxn],q[maxn],s1[maxn],s2[maxn];
int a1,a2,a3,a4,a5,a6,a7,a8,a9,a10;
int i,j,k,l,r,t,n,m,inv2,inv6,inv100,ans;
int read(){
int x=0;
char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9'){
x=x*10+ch-'0';
ch=getchar();
}
return x;
}
int quicksortmi(int x,int y){
if (!y) return 1;
int t=quicksortmi(x,y/2);
t=(ll)t*t%mo;
if (y%2) t=(ll)t*x%mo;
return t;
}
int main(){
freopen("sequence.in","r",stdin);freopen("sequence.out","w",stdout);
scanf("%d",&n);
inv2=quicksortmi(2,mo-2);inv6=quicksortmi(6,mo-2);inv100=quicksortmi(100,mo-2);
fo(i,1,n){
t=read();
p[i]=(ll)t*inv100%mo;q[i]=(ll)(100-t)*inv100%mo;
}
fo(i,0,1){
s1[i]=(ll)i*(i+1)%mo*inv2%mo;
s2[i]=(ll)s1[i]*s1[i]%mo;
}
q[0]=1;
a2=(ll)s2[1]*p[1]%mo;
a4=(q[1]+p[1])%mo;
a5=-p[1];
a6=p[1];
a7=-p[1];
a10=(ll)s1[1]*p[1]%mo;
fo(i,1,n){
g[i]=(a8+a10)%mo;
f[i]=((ll)a1+(ll)a2+(ll)a3*2%mo)%mo;
a1=((ll)a1*p[i+1]%mo+(ll)f[i]*q[i+1]%mo)%mo;
a2=((ll)a2+(ll)i*i%mo*i%mo*a4%mo-(ll)3*i%mo*i%mo*a5%mo+(ll)3*i%mo*a6%mo-(ll)a7)%mo;
//a2=((ll)a2*p[i+1]%mo+(ll)s2[0]*q[i+1]%mo)%mo;
a2=(ll)a2*p[i+1]%mo;
a3=((ll)a3+(ll)i*a8%mo-(ll)a9)%mo;
//a3=((ll)a3*p[i+1]%mo+(ll)s1[0]*g[i]%mo*q[i+1]%mo)%mo;
a3=(ll)a3*p[i+1]%mo;
a10=((ll)a10+(ll)i*a4%mo-(ll)a5)%mo;
//a10=((ll)a10*p[i+1]%mo+s1[0]*q[i+1]%mo)%mo;
a10=(ll)a10*p[i+1]%mo;
a4=((ll)a4*p[i+1]%mo+(ll)q[i+1])%mo;
a5=((ll)a5*p[i+1]%mo+(ll)q[i+1]*i%mo)%mo;
a6=((ll)a6*p[i+1]%mo+(ll)q[i+1]*i%mo*i%mo)%mo;
a7=((ll)a7*p[i+1]%mo+(ll)q[i+1]*i%mo*i%mo*i%mo)%mo;
a8=((ll)a8*p[i+1]%mo+(ll)q[i+1]*g[i]%mo)%mo;
a9=((ll)a9*p[i+1]%mo+(ll)q[i+1]*g[i]%mo*i%mo)%mo;
}
ans=f[n];
(ans+=mo)%=mo;
printf("%d\n",ans);
//printf("%d\n",(ll)61*quicksortmi(8,mo-2)%mo);
}