牛客网暑期ACM多校训练营（第五场）F take 线段树 概率

题目链接: F take

题目大意

有n个箱子, 每个箱子里面有p[i]/100的概率有一个大小为d[i]的钻石
一开始你手上的钻石大小为0, 你从第一个箱子开始, 依次打开每一个箱子, 如果箱子里面的钻石大小比你手上的大, 那就拿起箱子里的钻石替换自己的, 求最后替换次数的期望

思路

考虑第i次开箱子, 箱子里钻石比你手上的大的概率为, 当前钻石比d[i]小的概率*开箱子开出钻石的概率
$P_i = \sum_{size=0}^{d[i]-1}p(当前钻石大小为size) \cdot \frac{p[i]}{100}$
所以第i次开箱子对总答案的贡献是$1 \cdot P_i$
最后的答案是$E= \sum_{i=1}^{n}P_i$
然后就是怎么记录当前钻石比箱子里的钻石小的概率了
将钻石大小离散化, 建一个线段树, 线段树每个节点保存的是当前钻石大小为i的概率, 这样就可以快速求出当前钻石大小为0到d[i]-1的概率
每次开箱子之后, 将区间[0, d[i]-1]的所有节点概率乘以一个(100-p[i])/100, 也就是没有开出钻石的概率(只有没有开出钻石, 手上钻石大小才可能不变), 然后将位置d[i]的概率加上一个
$P_i = \sum_{size=0}^{d[i]-1}p(当前钻石大小为size) \cdot \frac{p[i]}{100}$, 也就是手上钻石被替换成箱子里的钻石的概率
线段树有两个更新函数, 一个是将区间乘以一个值, 要用懒惰标记, 一个是将一个点加上一个值
注意mod, 所有地方不要忘记加mod, 除100变成乘以100的逆元

代码

运行时间： 246 ms 占用内存：7788K

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int maxn = 1e5 + 100;
const ll mod = 998244353;
ll n, p[maxn], d[maxn], all[maxn];

ll inv[maxn];


#define ls l, m, rt<<1
#define rs m+1, r, rt<<1|1
#define defm int m = (l+r)>>1

ll sum[maxn << 2], col[maxn << 2];

void pushUp(int rt)
{
    sum[rt] = (sum[rt << 1] + sum[rt << 1 | 1])%mod;
}
void pushDown(int rt)
{
    if (col[rt] != 1)
    {
        col[rt << 1] *= col[rt];
        col[rt << 1] %= mod;
        col[rt << 1 | 1] *= col[rt];
        col[rt << 1 | 1] %= mod;
        sum[rt << 1] *= col[rt];
        sum[rt << 1] %= mod;
        sum[rt << 1 | 1] *= col[rt];
        sum[rt << 1 | 1] %= mod;
        col[rt] = 1;
    }
}
void build(int l, int r, int rt)
{
    col[rt] = 1;
    sum[rt] = 0;
    if (l == r) return ;
    int m = (l + r) / 2;
    build(ls);
    build(rs);
}
void update(int L, int R, ll c, int l, int r, int rt)
{
    if (L <= l && r <= R)
    {
        col[rt] *= c;
        col[rt] %= mod;
        sum[rt] *= c;
        sum[rt] %= mod;
        return ;
    }
    pushDown(rt);
    int m = (l + r) / 2;
    if (L <= m) update(L, R, c, ls);
    if (R > m) update(L, R, c, rs);
    pushUp(rt);
}
void add(int p, ll c, int l, int r, int rt)
{
    if (l == r)
    {
        sum[rt] += c;
        sum[rt] %= mod;
        return ;
    }
    pushDown(rt);
    int m = (l + r) / 2;
    if (p <= m) add(p, c, ls);
    if (p > m) add(p, c, rs);
    pushUp(rt);
}
ll query(int L, int R, int l, int r, int rt)
{
    if (L <= l && r <= R)
    {
        return sum[rt];
    }
    pushDown(rt);
    int m = (l + r) / 2;
    ll ret = 0;
    if (L <= m) ret += query(L, R, ls);
    if (m < R) ret += query(L, R, rs);
    return ret % mod;
}

int main()
{
    inv[1] = 1;
    for (int i = 2; i < maxn; ++i)
        inv[i] = inv[mod % i] * (mod - mod / i) % mod; //O(n)预处理逆元
    cin >> n;
    for (int i = 0; i < n; ++i)
    {
        cin >> p[i] >> d[i];
        all[i] = d[i];
    }
    sort(all, all + n);
    int len = unique(all, all + n) - all;
    for (int i = 0; i < n; ++i)
    {
        d[i] = lower_bound(all, all + len, d[i]) - all + 1;
    }

    build(0, len + 1, 1);
    add(0, 100 * inv[100] % mod, 0, len + 1, 1);
    ll ans = 0;
    for (int i = 0; i < n; ++i)
    {
        ll tp = p[i] * inv[100] % mod * query(0, d[i] - 1, 0, len + 1, 1) % mod;
        ans += tp;
        ans %= mod;
        update(0, d[i] - 1, (100 - p[i])*inv[100] % mod, 0, len + 1, 1);
        add(d[i], tp, 0, len + 1, 1);
    }
    cout << ans << endl;

    return 0;
}