牛客网wannafly22 D 整数序列

/*
  Source   :wannafly 22 D 整数序列
  Problem  :给定长度为n的数组，有m次操作，op=1 时， 把a[l-r] 中的数加v, op=2时， 求sum(sin(a[l-r]))
  Solution :由 sin(a+b) = sin(a)cos(b) + cos(a)*sin(b)
               cos(a+b) = cos(a)cos(b) + sin(a)*sin(b)
            可以维护区间的sin和以及cos和。
  Date     :2018-08-17-18.48
*/

#include <bits/stdc++.h>
#include <cmath>
using namespace std;

typedef long long LL;
const int MAXN = 200005;
const LL MOD7 = 1e9+7;

int read()
{
    int x=0;char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}

struct Node
{
    int l,r;
    double c;   //这里可能会爆int
    double a;   //sin
    double b;   //cos
};

struct SegTree
{
    Node f[MAXN*8];
    void pushUp(int t)
    {
        f[t].a=f[2*t].a+f[2*t+1].a;
        f[t].b=f[2*t].b+f[2*t+1].b;
        // f[t].g=f[2*t].g+f[2*t+1].g;
    }

    void pushDown(int t)
    {
        if (f[t].c>1e-9)
        {
            f[2*t].c+=f[t].c;
            double cosv = cos(f[t].c);     // 这里优化cos和sin的计算次数
            double sinv = sin(f[t].c);
            double tmp = f[2*t].a * cosv + f[2*t].b * sinv;
            f[2*t].b = -f[2*t].a * sinv + f[2*t].b * cosv;
            f[2*t].a = tmp;

            f[2*t+1].c+=f[t].c;
            tmp = f[2*t+1].a * cosv + f[2*t+1].b * sinv;
            f[2*t+1].b = -f[2*t+1].a * sinv + f[2*t+1].b * cosv;
            f[2*t+1].a = tmp;

            f[t].c=0;
        }
    }

    void build(int t,int l,int r, int *a)
    {
        f[t].l=l;
        f[t].r=r;
        f[t].c=0;
        f[t].a=0;
        f[t].b=0;
        if (l==r)
        {
            f[t].a=sin(1.0*a[l]);
            f[t].b=cos(1.0*a[l]);
            return;
        }
        int mid=(l+r)/2;
        build(2*t,l,mid,a);
        build(2*t+1,mid+1,r,a);
        pushUp(t);
    }

    void update(int t,int l,int r,int c)
    {
        // printf("update l=%d r=%d c=%d\n",l,r,c);
        if (f[t].l==l && f[t].r==r)
        {
            f[t].c+=c;
            double cosv = cos(c);
            double sinv = sin(c);
            double tmp = f[t].a * cosv + f[t].b * sinv;
            f[t].b = -f[t].a *  sinv  + f[t].b * cosv;
            f[t].a = tmp;
            return ;
        }
        pushDown(t);
        int mid=(f[t].l+f[t].r)/2;
        if (r<=mid) update(2*t,l,r,c);
        else if (l>mid) update(2*t+1, l,r, c);
        else
        {
            update(2*t,l,mid,c);
            update(2*t+1,mid+1,r,c);
        }
        pushUp(t);
    }

    double query(int t,int l,int r)
    {
        // printf("query t=%d l=%d r=%d\n",t,l,r);
        if (f[t].l==l && f[t].r==r)
        {
            return f[t].a;
        }
        int mid=(f[t].l+f[t].r)/2;
        pushDown(t);
        double ans=0;
        if (r<=mid) ans=query(2*t,l,r);
        else if (l>mid) ans=query(2*t+1,l,r);
        else ans=query(2*t,l,mid)+query(2*t+1,mid+1,r);
        return ans;
    }

    void Debug(int t)
    {
        printf("************tree Node**********\n");
        printf("\t l=%d r=%d \t\n", f[t].l, f[t].r);
        printf("\t c=%d \t\n",f[t].c);
        printf("\t a=%f \t\n",f[t].a);
        printf("\t b=%f \t\n",f[t].b);
        // printf("\t g=%f \t\n",f[t].g);
        printf("************END NODE**********\n");
        if (f[t].l==f[t].r) return;
        Debug(2*t);
        Debug(2*t+1);
    }
}tree;

int a[MAXN];
int n,m;
int x;

int main()
{
    // printf("round(-1.37)=%.1f\n", -1.37);
#ifndef ONLINE_JUDGE
    freopen("test.txt","r",stdin);
#endif // ONLINE_JUDGE
    n=read();
    for (int i=1;i<=n;++i)
    {
        a[i]=read();
    }
    tree.build(1,1,n,a);
    // tree.Debug(1);
    m=read();
    int op,l,r;
    for (int i=1;i<=m;++i)
    {
        op=read();
        l=read();
        r=read();
        // scanf("%d%d%d",&op,&l,&r);
        // printf("op=%d l=%d r=%d\n",op,l,r);
        if (op==1)
        {
            // scanf("%d",&x);
            x=read();
            tree.update(1,l,r,x);
            // tree.Debug(1);
        }
        else
        {
            double tmp = tree.query(1, l, r);
            printf("%.1f\n",tmp);
            // tree.Debug(1);
        }
    }
    return 0;
}