6028: 白日梦（DP，期望=∑概率*数量）

6028: 白日梦

时间限制: 1 Sec   内存限制: 128 MB
提交: 47   解决: 26
[ 提交][ 状态][ 讨论版][命题人: admin]

题目描述

SR需要相当大的睡眠量
某日，他做了一个奇怪的梦，他梦见自己成为了怪物猎人，为了狩猎，他要去一个岛上住N+1天（编号为0到N）。这个岛位于太平洋中心，每天要么是晴天，要么刮台风。
他到达岛的第0天是晴天。然后对于第i天，假如是晴天，那么有P(0<p<=1)的概率会变天，使得接下来连续M天都刮台风，然后第i+M+1天必然会转晴。
天气对SR的狩猎造成了很大的影响，如果第i天是晴天，那么SR能狩猎到A只猎物；如果刮台风，SR只能狩猎到B只猎物，还有D(0<D<=1)的概率会失败，在失败的情况下，只能狩猎到C只猎物
SR突然幻梦初醒，他想知道编号1到N天的狩猎到的猎物数量和的期望值
设答案为x/y（x与y互素），请你输出x*y^(-1)除以998244353的余数。这里y^(-1)指y的乘法逆元，它是1到998244353-1中的某一个整数，满足y^(-1)*y-1能被998244353整除（题目保证这个数存在）。

输入

一行7个整数N,M,P,D,A,B,C，如题目所描述（其中给出的P，D是模998244353意义下的值）

输出

一个整数，为答案。

样例输入

3 1 499122177 499122177 1 2 3

样例输出

311951365

提示

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

提示
样例一输出的值对应的实数是4.6875

来源

2018江苏冬令营5 

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define mod 998244353
using namespace std;
long long ans=0, m, n, p, d, a, b, c, dp[1000006];
int main()
{
    dp[0]=1;
    scanf("%lld%lld%lld%lld%lld%lld%lld", &n, &m, &p, &d, &a, &b, &c);
    for(int i=1;i<=n;i++)
    {
        dp[i]=dp[i-1]*(1-p+mod)%mod;
        if(i-m-1>=0)
            dp[i]=(dp[i]+dp[i-m-1]*p%mod)%mod;
     
        ans+=dp[i]*a%mod;
        ans+=(1-d+mod)%mod*(1-dp[i]+mod)%mod*b%mod;//注意：这里不能把(1-dp[i]+mod)%mod放在前面
        ans+=d*(1-dp[i]+mod)%mod*c%mod;
    }
    printf("%lld\n", (ans+mod)%mod);
    return 0;
} 
/**************************************************************
    Problem: 6028
    User: ldu_reserver201701
    Language: C++
    Result: 正确
    Time:52 ms
    Memory:8904 kb
****************************************************************/