[Wannafly挑战赛22] D 整数序列 线段树

题目描述 

给出一个长度为n的整数序列a1,a2,...,an，进行m次操作，操作分为两类。
操作1：给出l,r,v，将al,al+1,...,ar分别加上v；
操作2：给出l,r，询问

输入描述:

第一行一个整数n
接下来一行n个整数表示a1,a2,...,an
接下来一行一个整数m
接下来m行，每行表示一个操作，操作1表示为1 l r v，操作2表示为2 l r
保证1≤n,m,ai,v≤200000；1≤l≤r≤n，v是整数

输出描述:

对每个操作2，输出一行，表示答案，四舍五入保留一位小数
保证答案的绝对值大于0.1，且答案的准确值的小数点后第二位不是4或5
数据随机生成（n,m人工指定，其余整数在数据范围内均匀选取），并去除不满足条件的操作2

示例1

输入

复制

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

---

题意

n个数的序列，实现两种操作 1 l r v 将[l,r]区间加上v 

2 l r 查询[l,r]区间的sigma(sin(ai))

---

题解

肯定会想到用线段树维护这样的操作，可是sin(xi)并不满足区间可加性。

转换一下

sin(x+v) = sin(x)*cos(v) + cos(x)*sin(v)

cos(x+v) = cos(x)*cos(v) - sin(x)*sin(v)

我们只需要维护sin(x) 和 cos(x) 就可以轻易的得到sin(x+v) 和 cos(x+v)

如果引入复数的：

cos(x)+sin(x)i * cos(v)+sin(v)i = cos(x)*cos(v)-sin(x)*sin(v)+cos(x)*sin(v)+sin(x)*cos(v) = cos(x+v)+sin(x+v)i

---

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
typedef long long ll;
typedef complex<double> dcom;
struct SegTree {
    dcom sum[maxn<<2],lz[maxn<<2];
    void push_up(int rt) {
        sum[rt] = sum[rt<<1]+sum[rt<<1|1];
    }
    void push_down(int rt) {
        if(lz[rt] == dcom(1,0)) return;
        lz[rt<<1] *= lz[rt];lz[rt<<1|1] *= lz[rt];
        sum[rt<<1] *= lz[rt];
        sum[rt<<1|1] *= lz[rt];
        lz[rt] = dcom(1,0);
    }
    void build(int l,int r,int rt) {
        lz[rt] = dcom(1,0);
        if(l == r) {
            ll x;scanf("%lld",&x);
            sum[rt] = dcom(cos(x),sin(x));
            return;
        }
        int mid = (l + r) >> 1;
        build(l,mid,rt<<1);
        build(mid+1,r,rt<<1|1);
        push_up(rt);
    }
    void update(int ql,int qr,int l,int r,dcom val,int rt) {
        if(ql == l && qr == r) {
            lz[rt] *= val;
            sum[rt] *= val;
            return;
        }
        push_down(rt);
        int mid = (l + r) >> 1;
        if(ql <= mid) update(ql,qr,l,mid,val,rt<<1);
        if(qr > mid) update(ql,qr,mid+1,r,val,rt<<1|1);
        else {
            update(ql,mid,l,mid,val,rt<<1);
            update(mid+1,qr,mid+1,r,val,rt<<1|1);
        }
        push_up(rt);
    }
    dcom query(int ql,int qr,int l,int r,int rt) {
        if(ql == l && qr == r) return sum[rt];
        push_down(rt);
        int mid = (l + r) >> 1;
        if(qr <= mid) return query(ql,qr,l,mid,rt<<1);
        if(ql > mid) return query(ql,qr,mid+1,r,rt<<1|1);
        return query(ql,mid,l,mid,rt<<1)+query(mid+1,qr,mid+1,r,rt<<1|1);
    }
}seg;
int n,m;
int main()
{
    scanf("%d",&n);
    seg.build(1,n,1);
    scanf("%d",&m);
    int op,l,r;ll val;
    while(m--) {
        scanf("%d%d%d",&op,&l,&r);
        if(op == 1) {
            scanf("%lld",&val);
            dcom tmp(cos(val),sin(val));
            seg.update(l,r,1,n,tmp,1);
        }
        else {
            printf("%.1f\n",(double)seg.query(l,r,1,n,1).imag());
        }
    }
    return 0;
}