jzoj 6271. 2019.8.4【NOIP提高组A】锻造 (forging)

Description

详见OJ

Solution

这题看题目就知道是期望(DP)了。
先刚了2h(DP)式，得到(f[i]=f[i-1]+f[i-2]+f[i-1]*(1-p)+...)，然后不会化简，最后崩盘。
正解也是设f[i]表示生成第i级的剑的期望费用。
可以得到(f[i] = f[i - 1] + f[i - 2] + (1 - p) * (f[i] - f[i - 2]))
解方程即可，得到：
[f[i] = f[i - 2] + f[i - 1] * (k/c[i - 1])]（(k)为题目所给的(k)）
要用逆元（考场逆元直接用成了(/(k!))，而需要的是(/k)。。。）
[f[i] = f[i - 2] + f[i - 1] * k * ny[c[i - 1]] * jc[c[i - 1] - 1]]
此题需要卡常！！！

Code

#include <cstdio>
#include <algorithm>
#define N 10000010
#define ll long long
#define mo 998244353
#define fo(x, a, b) for (register int x = a; x <= b; x++)
#define fd(x, a, b) for (register int x = a; x >= b; x--)
using namespace std;
const int maxn = 10000000;
int n, A, bx, by, cx, cy, p, k;
int b[N], c[N], f[N], jc[N], ny[N];

inline int read()
{
    int x = 0; char c = getchar();
    while (c < '0' || c > '9') c = getchar();
    while (c >= '0' && c <= '9') x = (x << 1) + (x << 3) + (c ^ 48), c = getchar();
    return x;
}

int ksm(int x, int y)
{
    int s = 1;
    while (y)
    {
        if (y & 1) s = (ll)s * x % mo;
        x = (ll)x * x % mo; y >>= 1;
    }
    return s;
}

void ycl()
{
    jc[0] = jc[1] = 1;
    fo(i, 2, maxn) jc[i] = (ll)jc[i - 1] * i % mo;
    ny[maxn] = ksm(jc[maxn], mo - 2);
    fd(i, maxn - 1, 1) ny[i] = (ll)ny[i + 1] * (i + 1) % mo;
}

int main()
{
    freopen("forging.in", "r", stdin);
    freopen("forging.out", "w", stdout);
    n = read(), A = read();
    if (n == 0) {printf("%d
", A); return 0;} ycl();
    bx = read(), by = read(), cx = read(), cy = read(), p = read();
    b[0] = by + 1, c[0] = cy + 1;
    fo(i, 1, n - 1)
    {
        b[i] = ((ll)b[i - 1] * bx + by) % p + 1;
        c[i] = ((ll)c[i - 1] * cx + cy) % p + 1;
    }
    f[0] = A; k = std::min(c[0], b[0]);
    f[1] = (f[0] + (ll)f[0] * c[0] % mo * ny[k] % mo * jc[k - 1] % mo) % mo;
    fo(i, 2, n)
    {
        k = std::min(c[i - 1], b[i - 2]);
        f[i] = (f[i - 2] + (ll)f[i - 1] * c[i - 1] % mo * ny[k] % mo * jc[k - 1] % mo) % mo;
    }
    printf("%d
", f[n]);
    return 0;
}