题目链接: F take
题目大意
有n个箱子, 每个箱子里面有p[i]/100的概率有一个大小为d[i]的钻石
一开始你手上的钻石大小为0, 你从第一个箱子开始, 依次打开每一个箱子, 如果箱子里面的钻石大小比你手上的大, 那就拿起箱子里的钻石替换自己的, 求最后替换次数的期望
思路
考虑第i次开箱子, 箱子里钻石比你手上的大的概率为, 当前钻石比d[i]小的概率*开箱子开出钻石的概率
Pi=∑d[i]−1size=0p(当前钻石大小为size)⋅p[i]100
P
i
=
∑
s
i
z
e
=
0
d
[
i
]
−
1
p
(
当
前
钻
石
大
小
为
s
i
z
e
)
⋅
p
[
i
]
100
所以第i次开箱子对总答案的贡献是
1⋅Pi
1
⋅
P
i
最后的答案是
E=∑ni=1Pi
E
=
∑
i
=
1
n
P
i
然后就是怎么记录当前钻石比箱子里的钻石小的概率了
将钻石大小离散化, 建一个线段树, 线段树每个节点保存的是当前钻石大小为i的概率, 这样就可以快速求出当前钻石大小为0到d[i]-1的概率
每次开箱子之后, 将区间[0, d[i]-1]的所有节点概率乘以一个(100-p[i])/100, 也就是没有开出钻石的概率(只有没有开出钻石, 手上钻石大小才可能不变), 然后将位置d[i]的概率加上一个
Pi=∑d[i]−1size=0p(当前钻石大小为size)⋅p[i]100
P
i
=
∑
s
i
z
e
=
0
d
[
i
]
−
1
p
(
当
前
钻
石
大
小
为
s
i
z
e
)
⋅
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;
}