刷题记录:牛客NC17877整数序列

最新推荐文章于 2024-07-08 22:12:26 发布

yingjiayu12

最新推荐文章于 2024-07-08 22:12:26 发布

阅读量367

点赞数

分类专栏： c++算法文章标签：算法 c++ 数据结构线段树

本文链接：https://blog.csdn.net/yingjiayu12/article/details/128941485

版权

c++算法专栏收录该内容

425 篇文章 1 订阅

订阅专栏

传送门:牛客

题目描述:

给出一个长度为n的整数序列a1,a2,...,an，进行m次操作，操作分为两类。
操作1：给出l,r,v，将al,al+1,...,ar分别加上v；
操作2：给出l,r，询问sin(al)+...+sin(ar)
输入:
4
1 2 3 4
5
2 2 4
1 1 3 1
2 2 4
1 2 4 2
2 1 3
输出:
0.3
-1.4
-0.3

一道线段树维护区间修改的题目,但是此时变成了区间 $s in$ 和

但是没有关系,当你完全明白如何维护区间和的时候,其实区间 $s in$ 和也就不难了

对于每一个区间 $[l, r]$ 里面的值 $a_i$ ,我们给他加上一个值 $v$ ,此时就有:
$\sum_{l}^r{sin(a_i+v)}=\sum_{l}^r{(sina_icosv+cosa_isinv)}$
$=cosv\sum{sina_i}+sinv\sum{cosa_i}$
我们发现与之前的区间 $s in$ 和以及 $cos$ 和有关.那么此时我们就用线段树同时来维护区间 $s in$ 和,区间 $cos$ 和.维护方式就与维护区间加一样使用懒标记即可.
对于区间 $cos$ 和,我们用类似于上述的推导方式也会得到一个类似的关于 $s in$ 和与 $cos$ 和的式子:
$\sum_{l}^r{cos(a_i+v)}=\sum_{l}^r{(cosa_icosv-sina_isinv)}$
$=cosv\sum_{l}^r{cosa_i}-sinv\sum_{l}^r{sina_i}$

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
struct Segment_tree{
	int l,r;double Sum_sin,Sum_cos,lazy;
}tree[maxn*4];
int n,m;int a[maxn];
void pushup(int rt) {
	tree[rt].Sum_sin=tree[ls].Sum_sin+tree[rs].Sum_sin;
	tree[rt].Sum_cos=tree[ls].Sum_cos+tree[rs].Sum_cos;
}
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;
	if(l==r) {
		tree[rt].Sum_sin=sin((double)a[l]);
		tree[rt].Sum_cos=cos((double)a[l]);
		return ;
	} 
	int mid=(l+r)>>1;
	build(lson);build(rson);
	pushup(rt);
}
void change(int rt,double v) {
	tree[rt].lazy+=v;
	double Sums=tree[rt].Sum_sin,Sumc=tree[rt].Sum_cos;
	tree[rt].Sum_sin=Sums*cos(v)+Sumc*sin(v);
	tree[rt].Sum_cos=Sumc*cos(v)-Sums*sin(v);
}
void pushdown(int rt) {
	change(ls,tree[rt].lazy);change(rs,tree[rt].lazy);
	tree[rt].lazy=0;
}
void update(int l,int r,int rt,double v) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		change(rt,v);
		return ;
	}
	if(tree[rt].lazy) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) update(l,r,ls,v);
	else if(l>mid) update(l,r,rs,v);
	else update(l,mid,ls,v),update(mid+1,r,rs,v);
	pushup(rt);
}
double query(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		return tree[rt].Sum_sin;
	}
	if(tree[rt].lazy) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) return query(l,r,ls);
	else if(l>mid) return query(l,r,rs);
	else return query(l,mid,ls)+query(mid+1,r,rs);
}
int main() {
	n=read();
	for(int i=1;i<=n;i++) a[i]=read();
	build(root);
	m=read();
	for(int i=1;i<=m;i++) {
		int opt=read();
		if(opt==1) {
			int l=read(),r=read(),v=read();
			update(l,r,1,(double)v);
		}
		else {
			int l=read(),r=read();
			printf("%.1f\n",query(l,r,1));
		}
	}
	return 0;
}