luogu P6327 区间加区间sin和 |线段树

题目描述

给出一个长度为 (n) 的整数序列 (a_1,a_2,ldots,a_n),进行 (m) 次操作,操作分为两类。

操作 (1):给出 (l,r,v),将 (a_l,a_{l+1},ldots,a_r)​ 分别加上 (v)

操作 (2):给出 (l,r),询问 (sumlimits_{i=l}^{r}sin(a_i))

输入格式

第一行一个整数 (n)

接下来一行 (n) 个整数表示 (a_1,a_2,ldots,a_n)​。

接下来一行一个整数 (m)

接下来 (m) 行,每行表示一个操作,操作 (1) 表示为 1 l r v,操作 (2) 表示为 2 l r。

输出格式

对每个操作 (2),输出一行,表示答案,四舍五入保留一位小数。

保证答案的绝对值大于 (0.1),且答案的准确值的小数点后第二位不是 (4)(5)


#include <bits/stdc++.h>
using namespace std;
const int N=2e5+5,mod=1e9+7;
#define int long long
inline int read(){
	int x=0,f=1; char c=getchar();
	while(c<'0'||c>'9'){ if(c=='-')f=-1;  c=getchar(); }
	while('0'<=c&&c<='9'){ x=(x<<1)+(x<<3)+(c^48); c=getchar();  }
	return x*f;	
}
#define db double
struct node{
	int l,r,add;
	db sinv,cosv;	
	#define l(x) T[x].l
	#define r(x) T[x].r
	#define add(x) T[x].add
	#define s(x) T[x].sinv
	#define c(x) T[x].cosv
}T[N<<2];
#define ls p<<1
#define rs p<<1|1
#define mid ((l(p)+r(p))>>1)
int n,a[N];
void build(int p,int l,int r){
	l(p)=l,r(p)=r;
	if(l==r){ s(p)=sin(a[l]),c(p)=cos(a[l]); return; }
	build(ls,l,mid),build(rs,mid+1,r);
	s(p)=s(ls)+s(rs),c(p)=c(ls)+c(rs);
}
void around(int p,db sinx,db cosx){
	db S=s(p),C=c(p);
	s(p)=cosx*S+sinx*C;
	c(p)=cosx*C-sinx*S;
}
void pushdown(int p){
	around(ls,sin(add(p)),cos(add(p)));
	around(rs,sin(add(p)),cos(add(p)));
	add(ls)+=add(p);
	add(rs)+=add(p);
	add(p)=0;
}
void update(int p,int l,int r,int d){
	if(l<=l(p)&&r(p)<=r){
		around(p,sin(d),cos(d));
		add(p)+=d;
		return;	
	}
	if(add(p))pushdown(p);
	if(l<=mid)update(ls,l,r,d);
	if(r>mid)update(rs,l,r,d);
	s(p)=s(ls)+s(rs),c(p)=c(ls)+c(rs);
}
db query(int p,int l,int r){
	if(l<=l(p)&&r(p)<=r)return s(p);
	if(add(p))pushdown(p);
	db ans=0;
	if(l<=mid)ans+=query(ls,l,r);
	if(r>mid)ans+=query(rs,l,r);
	return ans;
}
signed main(){
	n=read();
	for(int i=1;i<=n;i++)a[i]=read();
	build(1,1,n);
	int m=read();
	int l,r,opt;
	while(m--){
		opt=read(),l=read(),r=read();
		if(opt==1){
			int v=read();
			update(1,l,r,v);
		}else printf("%.1lf
",query(1,l,r));	
	}
}
原文地址:https://www.cnblogs.com/naruto-mzx/p/13049980.html