【洛谷 P3772】[CTSC2017]游戏(矩阵乘法+线段树)

传送门

洛谷题解区牛逼啊
一个贴的 c z _   x u y i x u a n cz \_ \ xuyixuan cz_ xuyixuan的,一个贴的 D Z Y O DZYO DZYO
为了发题解连亲妈都不要了吗?


首先显然考虑每个位置赢的概率

问题在于
一个位置输赢的概率不只要考虑前面的
还要考虑造成下一个胜负确定位置的影响

参考给出的贝叶斯公式

P ( x i = 1 ∣ x r , x l ) = P ( x l , x r ∣ x i ) P ( x i ) P ( x l , x r ) P(x_i=1|x_r,x_l)=\frac{P(x_l,x_r|x_i)P(x_i)}{P(x_l,x_r)} P(xi=1xr,xl)=P(xl,xr)P(xl,xrxi)P(xi)
= P ( x l , x r , x i ) P ( x i ) P ( x i ) P ( x l ) P ( x r ∣ x l ) =\frac{\frac{P(x_l,x_r,x_i)}{P(x_i)}P(x_i)}{P(x_l)P(x_r|x_l)} =P(xl)P(xrxl)P(xi)P(xl,xr,xi)P(xi)
= P ( x l , x r , x i ) P ( x l ) P ( x r ∣ x l ) =\frac{P(x_l,x_r,x_i)}{P(x_l)P(x_r|x_l)} =P(xl)P(xrxl)P(xl,xr,xi)
= P ( x i , x r ∣ x l ) P ( x r ∣ x l ) =\frac{P(x_i,x_r|x_l)}{P(x_r|x_l)} =P(xrxl)P(xi,xrxl)

可以发现下面就是 x l x_l xl导致 x r x_r xr为当前确认情况的概率,这个就是概率矩阵相乘
可以发现这时候
上面则不需要考虑 x r x_r xr的影响了,上面所有i加起来就是赢的场数的期望,只需要在概率矩阵相乘的时候顺便算一个期望即可

然后就只需要在加入/删除确定时间的时候找一下前驱后继加加减减就可以了

#include<bits/stdc++.h>
using namespace std;
#define cs const
#define re register
#define pii pair<int,int>
#define fi first
#define se second
#define ll long long
#define pb push_back
cs int RLEN=1<<20|1;
inline char gc(){
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob)&&(ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob)?EOF:*ib++;
}
#define gc getchar
inline int read(){
	char ch=gc();
	int res=0;bool f=1;
	while(!isdigit(ch))f^=ch=='-',ch=gc();
	while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=gc();
	return f?res:-res;
}
template<class tp>inline void chemx(tp &a,tp b){a<b?a=b:0;}
template<class tp>inline void chemn(tp &a,tp b){a>b?a=b:0;}
int n,m;
cs int N=200005;
struct mat{
	double a[2][2];
	mat(){a[0][0]=a[1][0]=a[0][1]=a[1][1]=0;}
	friend inline mat operator +(cs mat &a,cs mat &b){
		mat c;
		for(int i=0;i<2;i++)
		for(int j=0;j<2;j++)
		c.a[i][j]=a.a[i][j]+b.a[i][j];
		return c;
	}
	friend inline mat operator *(cs mat &a,cs mat &b){
		mat c;
		for(int i=0;i<2;i++)
		for(int k=0;k<2;k++)
		for(int j=0;j<2;j++)
		c.a[i][j]+=a.a[i][k]*b.a[k][j];
		return c;
	}
};
double p[N],q[N];
int sta[N];
namespace Seg{
	#define lc (u<<1)
	#define rc ((u<<1)|1)
	#define mid ((l+r)>>1)
	struct data{
		mat a,b;
		inline void init(double p,double q){
			a.a[0][0]=1-q,a.a[0][1]=q,a.a[1][0]=1-p,a.a[1][1]=p;
			b.a[0][1]=q,b.a[1][1]=p;
		}
		friend inline data operator *(cs data &a,cs data &b){
			data c;
			c.a=a.a*b.a;
			c.b=a.a*b.b+a.b*b.a;
			return c;
		}
	}s[N<<2];
	inline void pushup(int u){
		s[u]=s[lc]*s[rc];
	}
	void build(int u,int l,int r){
		if(l==r){
			s[u].init(p[l],q[l]);return;
		}
		build(lc,l,mid),build(rc,mid+1,r);
		pushup(u);
	}
	data query(int u,int l,int r,int st,int des){
		if(st<=l&&r<=des)return s[u];
		if(des<=mid)return query(lc,l,mid,st,des);
		if(mid<st)return query(rc,mid+1,r,st,des);
		return query(lc,l,mid,st,des)*query(rc,mid+1,r,st,des);
	}
	inline double getans(int l,int r){
		data now=query(1,0,n+1,l+1,r);
		return now.b.a[sta[l]][sta[r]]/now.a.a[sta[l]][sta[r]];
	}
	#undef lc
	#undef rc
	#undef mid
}
char op[5];
map<int,int> vt;
#define It map<int,int>::iterator
int main(){
	#ifdef Stargazer
	freopen("lx.in","r",stdin);
	#endif
	n=read(),m=read();scanf("%s",op);
	scanf("%lf",&p[1]);
	for(int i=2;i<=n;i++)scanf("%lf%lf",&p[i],&q[i]);
	p[0]=q[0]=1,p[n+1]=q[n+1]=0;
	Seg::build(1,0,n+1);
	vt[0]=1,vt[n+1]=0,sta[0]=1,sta[n+1]=0;
	double ans=Seg::getans(0,n+1);
	while(m--){
		scanf("%s",op);
		if(op[0]=='a'){
			int pos=read(),c=read();
			It nxt=vt.lower_bound(pos),pre=nxt;pre--;
			vt[pos]=c,sta[pos]=c;
			ans-=Seg::getans(pre->fi,nxt->fi);
			ans+=Seg::getans(pre->fi,pos);
			ans+=Seg::getans(pos,nxt->fi);
		}
		else{
			int pos=read();
			It nxt=vt.upper_bound(pos),pre=nxt;pre--,pre--;
			ans+=Seg::getans(pre->fi,nxt->fi);
			ans-=Seg::getans(pre->fi,pos);
			ans-=Seg::getans(pos,nxt->fi);
			sta[pos]=0;pre++,vt.erase(pre);
		}
		printf("%.7lf\n",ans);
	}
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值