【JZOJ5244】【NOIP2017模拟8.8A组】Daydreamin

Description

worldwideD最近有午睡的习惯~
某日中午，他做了一个梦：梦见有一个怪人，她去一个岛上住N+1天（编号为0到N）。这是在大洋中的岛，每天要么是晴天，要么刮台风。
她到达岛的第0天是晴天（这样她才能上岸）。然后对于第i天，假如是晴天，那么有P（0

Data Constraint

30%：N≤20
50%：N≤2,000
100%：1≤M≤N≤1,000,000 1≤A,B,C≤1,000 1≤P,D＜998244353

Solution

我们计算每一天为晴天的概率f[i]，每次f[i+m+1]+=f[i]p，f[i+1]+=f[i](1-p)。

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const ll maxn=1e6+5,mo=998244353;
ll f[maxn];
ll n,m,p,d,a,b,c,i,t,j,k,l,x,y,z,ans;
int main(){
    freopen("daydream.in","r",stdin);freopen("daydream.out","w",stdout);
    scanf("%lld%lld%lld%lld%lld%lld%lld",&n,&m,&p,&d,&a,&b,&c);
    b=(c*d+(1-d+mo)%mo*b)%mo;
    f[0]=1;x=(1-p+mo)%mo;
    for (i=0;i<n;i++){
        if (i+m+1<=n)f[i+m+1]=(f[i+m+1]+f[i]*p%mo)%mo;
        f[i+1]=(f[i+1]+f[i]*x%mo)%mo;
    }
    for (i=1;i<=n;i++)
        ans=(ans+f[i]*a+(1-f[i]+mo)%mo*b)%mo;
    printf("%lld\n",ans);
}