牛客网暑期ACM多校训练营（第五场） - F take （树状数组）

https://www.nowcoder.com/acm/contest/143/F

 

题意：

一个人依次按顺序打开1-n号宝箱。若宝箱里有一颗比他手上还大的钻石，他会换走。

给你宝箱i里Di大的钻石出现的几率是Pi，问你交换的期望。

 

POINT：

总期望就等于每个宝箱被交换的期望和。

第i个宝箱被交换的概率就是 前面1到i-1个宝箱都没有出现【比它大的钻石】的概率。

你可以用线段树，写起来麻烦但是想起来简单。

也可以利用树状数组进行前缀积来做。有点像逆序对的做法。

但是，树状数组只能维护前缀积，不能维护其中一段的积。（除非没有0）。

比如树状数组维护a,b,c,d的积。 要求c,d的积，你不能用a*b*c*d/(a*b)来求。

因为若a=0或者b=0，这个结果就错了。

所以这题的离散化要倒着来。query(x)代表比x大或等于的钻石不出现的概率。

ps：这题的数据没有钻石相等的情况。

 

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

#define  LL long long
const LL N = 100000+3;
const LL inf = 0x3f3f3f3f;
const LL mod = 998244353;
LL qkm(LL base,LL mi)
{
	LL ans=1;
	while(mi){
		if (mi&1)
			ans=ans*base%mod;
		base=base*base%mod;
		mi>>=1;
	}
	return ans;
}
LL mul[N];
LL n;
void add(LL x,LL p)
{
	for(LL i=x;i<=n;i+=i&-i)
		mul[i]=mul[i]*p%mod;
}

LL query(LL x)
{
	LL ans=1;
	for(LL i=x;i>=1;i-=i&-i)
		ans=ans*mul[i]%mod;
	return ans;
}
LL p[N],d[N];

vector<LL>V;

LL name(LL x)
{
	return lower_bound(V.begin(),V.end(),x)-V.begin()+1;
}

int main()
{
	LL inv=qkm(100,mod-2);
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld%lld",&p[i],&d[i]);
		V.push_back(d[i]);
	}
	sort(V.begin(), V.end());
	V.erase(unique(V.begin(),V.end()),V.end());
	LL ans=0;
	n=V.size();
	for(int i=1;i<=n;i++) mul[i]=1;

	for(int i=1;i<=n;i++){

		LL id = n-name(d[i])+1;
		ans+=((query(id)*p[i])%mod*inv)%mod;
		ans%=mod;
		LL x=(LL)(100-p[i])*inv%mod;
		add(id,x);
	}

	printf("%lld\n",ans);

	return 0;
}