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);
}