【JZOJ 4938】序列（70分做法）

最新推荐文章于 2019-12-24 17:32:26 发布

HOWARLI

最新推荐文章于 2019-12-24 17:32:26 发布

阅读量657

点赞数

分类专栏：概率与期望 DP 文章标签： DP 概率

本文链接：https://blog.csdn.net/HOWARLI/article/details/54428597

版权

DP 同时被 2 个专栏收录

50 篇文章 0 订阅

订阅专栏

概率与期望

9 篇文章 0 订阅

订阅专栏

Description

这里写图片描述

Solution

设DP方程： $f_i$ 表示做到i且i位为0的和期望， $g_i$ 表示做到i且i位为0的和期望平方，
设 $sum(i,j)=(i+j)*(i-j+1)/2$
先看一下f怎么求，

f i = \sum j = 0 i - 1 (f j + s u m (j + 1, i - 1)) * (1 - p j) * \prod k = j + 1 i - 1 p k

$f_i=\sum_{j=0}^{i-1} (f_j+sum(j+1,i-1))*(1-p_j)*\prod_{k=j+1}^{i-1}p_k$
这个很好理解，
有公式：

(a+b)2=a2+b2+2ab $(a+b)^2=a^2+b^2+2ab$
所有g的求法也和上面的差不多，就是加上了完全平方公式，

g i = \sum j = 0 i - 1 (g j + s u m (j + 1, i - 1) 2 + 2 s u m (j + 1, i - 1) * f j) * (1 - p j) * \prod k = j + 1 i - 1 p k

$g_i=\sum_{j=0}^{i-1} (g_j+sum(j+1,i-1)^2+2sum(j+1,i-1)*f_j)*(1-p_j)*\prod_{k=j+1}^{i-1}p_k$

最后输出 $g_{n+1}$

复杂度： $O(n^2)$ （辣鸡只会70分）

Code

#include <iostream>
#include <cstdio>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long LL;
const int N=2e6+50,mo=998244353;
int read(LL &n)
{
    char ch=' ';LL q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int n,m;
LL p[N],p1[N];
LL ps[N],psn[N];
LL f[N],g[N];
int ts[N];
LL ksm(LL q,LL w)
{
    LL a=q%mo,ans=1;
    while(w)
    {
        if(w%2) ans=(ans*a)%mo;
        a=(a*a)%mo;w=w/2;
    }
    return(ans);
}
int main()
{
    freopen("sequence.in","r",stdin);
    freopen("sequence.out","w",stdout);
    LL q,w,yb=ksm(100,mo-2);
    n=read(q);
    psn[0]=ps[0]=1;p[0]=0;p1[0]=1;
    fo(i,1,n)p1[i]=(100-read(p[i]))*yb%mo,p[i]=p[i]*yb%mo;
    p[n+1]=0;p1[n+1]=1;
    fo(i,1,n+1)
    {
        ts[i]=ts[i-1];
        if(p[i])ps[i]=p[i]*ps[i-1]%mo;else ps[i]=ps[i-1],ts[i]++;
        psn[i]=ksm(ps[i],mo-2);
    }
    ts[n+1]--;
    fo(i,1,n+1)
    {
        fo(j,0,i-1)if(ts[i-1]-ts[j]<1)
        {
            q=(i-j)*(i-j-1)/2%mo;w=(i-j)*(i-j-1)%mo;
            f[i]=(f[i]+(f[j]+q)*p1[j]%mo*(ps[i-1]*psn[j]%mo)%mo)%mo;
            g[i]=(g[i]+(g[j]+f[j]*w%mo+q*q%mo)%mo*p1[j]%mo*(ps[i-1]*psn[j]%mo)%mo)%mo;
        }
    }
    printf("%lld\n",g[n+1]);
    return 0;
}