codechef Chef at the Food Fair(线段树+泰勒展开)

传送门
题意:现在有 n n n个位置,每个位置上有一个值 a i a_i ai.
要求支持如下两种操作:

  1. 区间乘 v v v
  2. 求区间的 ( 1 − a i ) (1-a_i) (1ai)之积

思路:
考虑转换式子:
A n s = ∏ i = l r ( 1 − a i ) = e ∑ i = l r l n ( 1 − a i ) Ans=\prod_{i=l}^r(1-a_i)=e^{\sum_{i=l}^rln(1-a_i)} Ans=i=lr(1ai)=ei=lrln(1ai)
于是只需维护 ∑ i = l r l n ( 1 − a i ) \sum_{i=l}^rln(1-a_i) i=lrln(1ai)
把这个式子在 x 0 = 0 x_0=0 x0=0出泰勒展开可以得到: l n ( 1 − x ) = − x − 1 2 x 2 − 1 3 x 3 − ⋯ ˙ ln(1-x)=-x-\frac 12x^2-\frac 13x^3-\dot\cdots ln(1x)=x21x231x3˙
因为允许一定的精度误差。
于是我们维护这个多项式的前若干项即可。
代码:

#include<bits/stdc++.h>
#define ri register int
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
const int N=1e5+5;
double a[N];
int n,m;
namespace SGT{
	#define lc (p<<1)
	#define rc (p<<1|1)
	#define mid (T[p].l+T[p].r>>1)
	struct Val{
		double a[100];
		inline double&operator[](const int&k){return a[k];}
		inline const double&operator[](const int&k)const{return a[k];}
	};
	struct Tag{double a;};
	const Tag tag_empty=(Tag){1};
	Val val_empty;
	inline Val operator+(const Val&a,const Val&b){
		Val ret;
		for(ri i=0;i<100;++i)ret[i]=a[i]+b[i];
		return ret;
	}
	inline void operator+=(Val&a,const Val&b){a=a+b;}
	inline Tag operator+(const Tag&a,const Tag&b){return (Tag){a.a*b.a};}
	inline void operator+=(Tag&a,const Tag&b){a=a+b;}
	inline Val operator+(const Val&a,const Tag&b){
		Val ret;
		double mult=b.a;
		for(ri i=0;i<100;++i)ret[i]=a[i]*mult,mult*=b.a;
		return ret;
	}
	inline void operator+=(Val&a,const Tag&b){a=a+b;}
	inline Val solve1(double a){
		Val ret=val_empty;
		double mult=a;
		for(ri i=0;i<100;++i)ret[i]=-mult/(i+1),mult*=a;
		return ret;
	}
	inline double solve2(Val a){
		double ret=0.0;
		for(ri i=0;i<100;++i)ret+=a[i];
		return ret;
	}
	struct Node{Val val;Tag tag;int l,r;}T[N<<2];
	inline void pushup(int p){T[p].val=T[lc].val+T[rc].val;}
	inline void pushnow(int p,Tag v){T[p].val+=v,T[p].tag+=v;}
	inline bool check(Tag a){return fabs(a.a-1)<=1e-9;}
	inline void pushdown(int p){
		if(check(T[p].tag))return;
		pushnow(lc,T[p].tag),pushnow(rc,T[p].tag);
		T[p].tag=tag_empty;
	}
	inline void build(int p,int l,int r){
		T[p]=(Node){val_empty,tag_empty,l,r};
		if(l==r){T[p].val=solve1(a[l]);return;}
		build(lc,l,mid),build(rc,mid+1,r),pushup(p);
	}
	inline void update(int p,int ql,int qr,Tag v){
		if(ql<=T[p].l&&T[p].r<=qr)return pushnow(p,v);
		pushdown(p);
		if(ql<=mid)update(lc,ql,qr,v);
		if(qr>mid)update(rc,ql,qr,v);
		pushup(p);
	}
	inline Val query(int p,int ql,int qr){
		if(ql<=T[p].l&&T[p].r<=qr)return T[p].val;
		pushdown(p);
		Val ret=val_empty;
		if(ql<=mid)ret+=query(lc,ql,qr);
		if(qr>mid)ret+=query(rc,ql,qr);
		return ret;
	}
	#undef lc
	#undef rc
	#undef mid
}
int main(){
	for(ri i=0;i<100;++i)SGT::val_empty[i]=0;
	n=read(),m=read();
	for(ri i=1;i<=n;++i)scanf("%lf",&a[i]);
	SGT::build(1,1,n);
	double x;
	for(ri l,r,op;m;--m){
		op=read(),l=read(),r=read();
		if(!op)printf("%.8lf\n",exp(SGT::solve2(SGT::query(1,l,r))));
		else scanf("%lf",&x),SGT::update(1,l,r,(SGT::Tag){x});
	}
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值