P6327 区间加区间 sin 和 (线段树+数学)

传送门icon-default.png?t=N7T8https://www.luogu.com.cn/problem/P6327

 比较板子的一题,主要考察公式

//sin(a+x)=sinxcosa+sinacosx
//cos(a+x)=cosacosx-sinasinx

直接贴代码吧

// Problem: 
//     P6327 区间加区间 sin 和
//   
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P6327
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//sin(a+x)=sinxcosa+sinacosx
//cos(a+x)=cosacosx-sinasinx

#include<iostream>
#include<cmath>
#include<algorithm>
#include<iomanip>
using namespace std;
const int N=2e5+10;
#define lc u<<1
#define rc u<<1|1
double a[N];
struct Tree{
	int l,r;
	double ssum,csum,add;//无需维护val,这道题直接套公式即可
}tr[N*4];

void pushup(int u){
	tr[u].ssum=tr[lc].ssum+tr[rc].ssum;
	tr[u].csum=tr[lc].csum+tr[rc].csum;
}

void pushdown(int u){
	if(tr[u].add){
		tr[lc].add+=tr[u].add;
		tr[rc].add+=tr[u].add;
		double x1=tr[lc].ssum,y1=tr[lc].csum;
		tr[lc].ssum=(x1*cos(tr[u].add)+y1*sin(tr[u].add));
		tr[lc].csum=(y1*cos(tr[u].add)-x1*sin(tr[u].add));//可以推演一下多个的公式是一样的
		double x2=tr[rc].ssum,y2=tr[rc].csum;
		tr[rc].ssum=(x2*cos(tr[u].add)+y2*sin(tr[u].add));
		tr[rc].csum=(y2*cos(tr[u].add)-x2*sin(tr[u].add));
		tr[u].add=0;
	}
}

void build(int u,int l,int r){
	tr[u]={l,r};
	if(l==r){
		tr[u]={l,r,sin(a[l]),cos(a[l]),0};
		return;
	}
	int m=(l+r)>>1;
	build(lc,l,m);
	build(rc,m+1,r);
	pushup(u);
}

void update(int u,int l,int r,int k){
	if(l<=tr[u].l&&tr[u].r<=r){
		double x1=tr[u].ssum,y1=tr[u].csum;//别忘记开double,int会自动截断,这里调了好久
		tr[u].ssum=(x1*cos(k)+y1*sin(k));
		tr[u].csum=(y1*cos(k)-x1*sin(k));
		tr[u].add+=k;
		return;
	}
	pushdown(u);
	int m=(tr[u].l+tr[u].r)>>1;
	if(l<=m) update(lc,l,r,k);
	if(r>m) update(rc,l,r,k);
	pushup(u);
}

double query(int u,int l,int r){ 
	if(l<=tr[u].l&&tr[u].r<=r){
		return tr[u].ssum;
	}
	pushdown(u);
	int m=(tr[u].l+tr[u].r)>>1;
	double num=0;
	if(l<=m) num+=query(lc,l,r);
	if(r>m) num+=query(rc,l,r);
	return num;
}

int main(){
	int n;cin>>n;
	for(int i=1;i<=n;++i) cin>>a[i];
	build(1,1,n);
	int m;cin>>m;
	while(m--){
		int op;cin>>op;
		if(op==1){
			int x,y,z;cin>>x>>y>>z;
			update(1,x,y,z);
		}
		else{
			int x,y;cin>>x>>y;
			cout<<fixed<<setprecision(1)<<query(1,x,y)<<endl;
		}
	}
	//cout<<sin(10)<<endl;
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值