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\)，询问 \(\sum\limits_{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\n",query(1,l,r));	
	}
}