九校联考DAT2T1(dp,线性求逆元)

本文链接：https://blog.csdn.net/a1035719430/article/details/82622190

题目描述

“欢迎啊，老朋友。”

一阵寒暄过后，厂长带他们参观了厂子四周，并给他们讲锻造的流程。

“我们这里的武器分成若干的等级，等级越高武器就越厉害，并且对每一等级的武器都有两种属性值 b 和 c，但是我们初始只能花 a 个金币来生产 1 把 0 级剑……”

“所以你们厂子怎么这么垃圾啊，不能一下子就造出来 999 级的武器吗？”勇者不耐烦的打断了厂长的话。

“别着急，还没开始讲锻造呢……那我们举例你手中有一把 x 级武器和一把 y 级武器 (y = max(x − 1, 0))，我们令锻造附加值 k = min( $c_x$ , $b_y$ )，则你有 $\frac {k} {c_x}$ 的概率将两把武器融合成一把 x + 1 级的武器。”

“……但是，锻造不是一帆风顺的，你同样有1 - $\frac {k} {c_x}$ 的概率将两把武器融合成一把 max(x - 1, 0) 级的武器……”

勇者听完后暗暗思忖，他知道厂长一定又想借此机会坑骗他的零花钱，于是求助这个村最聪明的智者——你，来告诉他，想要强化出一把 n 级的武器，其期望花费为多少？

由于勇者不精通高精度小数，所以你只需要将答案对 998244353(7 ×17 × $2^{23}$ + 1，一个质数 ) 取模即可。

输入格式

第一行两个整数 n, a，含义如题所示。

为了避免输入量过大，第二行五个整数 $b_x, b_y, c_x, c_y, p$ ，按照下列代码来生成 b 和 c 数组。

b[0]=by+1;c[0]=cy+1;

for(int i=1;i

输出格式

输出一行一个整数，表示期望花费。

样例数据

input1

0 6432
4602677 3944535 2618884 6368297 9477531

output1

input2

1 3639650
6136976 5520115 2835750 9072363 9302097

output2

150643649

input3

10 2
2 33 6 66 2333333

output3

976750710

input4

200 5708788
0 0 0 0 1

output4

696441597

数据规模与约定

对于特殊性质处标示为“有”的数据满足 p = 1。

对于 100% 的数据，0 ≤ a ≤ $10^7$ , 0 ≤ $b_x, b_y, c_x, c_y$ < p < $10^7$ , 0 ≤ n ≤ $10^7$ 。

时间限制： $1 \text {s}$

空间限制： $256 \text {MB}$

看懂题目就很简单了
期望dp+线性求逆元

#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back 
#define pc putchar
#define mp make_pair
#define file(k) memset(k,0,sizeof(k))
#define ll long long
#define N 1e7+3
const int p = 998244353;
int n , a;
int b[10010000];
int c[10010000];
int inv[10010000]; 
int dp[10010000];
int read()
{
    int sum = 0;char c = getchar();bool flag = true;
    while(c < '0' || c > '9') {if(c == '-') flag = false;c = getchar();}
    while(c >= '0' && c <= '9') sum = sum * 10 + c - 48,c = getchar();
    if(flag) return sum;
    else return -sum;
}
void init()
{
    int bx = read() , by = read() , cx = read() , cy = read() , p = read();
    b[0] = by+1;c[0] = cy+1;
    rep(i,1,n-1) {
        b[i] = ((long long)b[i-1]*bx+by)%p+1;
        c[i] = ((long long)c[i-1]*cx+cy)%p+1;
    }
}
int mmin(int a,int b) {return a>b?b:a;}
void work()
{
    inv[1]=1;
    for(int i=2;i<N;i++)inv[i]=(ll)(p-p/i)*inv[p%i]%p;
    dp[0] = a;
    rep(i,1,n)
    {
        dp[i] = 1ll * dp[i-1] * c[i-1] % p;
        int k = mmin(c[i-1],b[i-2<0?0:i-2]);
        dp[i] = 1ll * dp[i] * inv[k]%p;
        dp[i] = (dp[i] + dp[i-2<0?0:i-2])%p;
    }
    printf("%d\n",dp[n]);
}
int main()
{
    freopen("forging.in","r",stdin);
    freopen("forging.out","w",stdout);
    n = read();a = read();
    init();
    work();
    return 0;
}