线段树练习题D

                                  Problem D: averyboy的午餐2

                                                  Time Limit: 2 Sec  Memory Limit: 128 MB
 

Description

因为聪明的你帮averyboy解决了荟园阿姨给他的问题，他顺利获得了免费的午餐。这次，averyboy再次来到了荟园，还是那个窗口，还是那位阿姨。。。。。。这次，阿姨不想那么容易让他吃到免费的午餐，所以给出了比上次要难的问题。这不，averyboy又来 求助你了。阿姨这次给出的问题如下.给你一个数组a[1]~a[N],然后给你q次操作，每次操作有三种类型，第一种类型是把数组中的某一个数改为x，第二种操作是询问区间[l, r]之间所有奇数的和，如果这个区间没有奇数，输出0，第三种操作是询问区间[l, r]之间所有偶数的和，如果这个区间没有偶数，输出0.

Input

第一行一个整数T，代表测试数据的组数.
接下来T组测试数据.
每组测试数据第一行两个整数N, q（N, q <= 1e5）,分别代表数组a的长度和操作的个数
接下来一行为N个整数，代表数组a
接下来q行，每一行格式为下面三种格式中的一种
1 loc x 将loc位置的数改为x，即a[loc] = x(1 <= loc <= N, 1 <= x <= 1e5)
2 l r 询问区间[l, r]之间奇数的和（1 <= l <= r <= N)
3 l r 询问区间[l, r]之间偶数的和（1 <= l <= r <= N)

 

Output

对于每组测试数据的询问操作，输出结果

解答：这个题目中维护的是奇数和偶数

#include<iostream>
using namespace std;
const long long  Max=100000;
struct Node
{
    int L,R;
    int Single;
    int Double;
}Node[Max<<2];
int a[Max+5];
void pushUp(int i)
{
    int lson = i<<1;
    int rson = lson + 1;
    Node[i].Single = Node[lson].Single + Node[rson].Single;
    Node[i].Double = Node[lson].Double + Node[rson].Double;
}
 
void build(int i, int l, int r)
{
    Node[i].L = l;
    Node[i].R = r;
    Node[i].Single=0;
    Node[i].Double=0;
    if(l == r&&a[l]%2==0)
    {
        Node[i].Double=a[l];
        Node[i].Single=0;
        return;
    }
    if(l == r&&a[l]%2!=0)
    {
        Node[i].Double=0;
        Node[i].Single=a[l];
        return;
    }
    int mid = (l + r)>>1;
    build(i<<1, l, mid);
    build((i<<1)|1, mid + 1, r);
    pushUp(i);
}
void update(int i,int loc,int value)
{
    if(Node[i].L == Node[i].R&&value%2==0)
    {
        Node[i].Double =value;
        Node[i].Single =0;
        return;
    }
      if(Node[i].L == Node[i].R&&value%2!=0)
    {
        Node[i].Double =0;
        Node[i].Single =value;
        return;
    }
    int mid = (Node[i].L + Node[i].R)>>1;
    if(loc <= mid) update(i<<1, loc, value);
    else update((i<<1)|1, loc, value);
    pushUp(i);
}
int querySingle(int i, int l, int r)
{
 
    if(Node[i].L == l && Node[i].R == r)
    {
        return Node[i].Single;
    }
    int mid = (Node[i].L + Node[i].R)>>1;
    if(r <= mid) return querySingle(i<<1, l, r);
    else if(l > mid) return querySingle((i<<1)|1, l, r);
    else
    {
        return querySingle(i<<1, l, mid) + querySingle((i<<1)|1, mid + 1, r);
    }
}
int queryDouble(int i, int l, int r)
{
 
    if(Node[i].L == l && Node[i].R == r)
    {
        return Node[i].Double;
    }
    int mid = (Node[i].L + Node[i].R)>>1;
    if(r <= mid) return queryDouble(i<<1, l, r);
    else if(l > mid) return queryDouble((i<<1)|1, l, r);
    else
    {
        return queryDouble(i<<1, l, mid) + queryDouble((i<<1)|1, mid + 1, r);
    }
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        int N,q;
        cin>>N>>q;
        for(int i=1;i<=N;i++)
        {
            cin>>a[i];
        }
        build(1,1,N);
        for(int i=1;i<=q;i++)
        {
            int m,p,q;
            cin>>m>>p>>q;
            if (m==1)
                {
            update(1, p, q);
                }
           if (m==2)
            {
                cout<<querySingle(1,p,q)<<endl;
            }
            if (m==3)
            {
                cout<<queryDouble(1,p,q)<<endl;
            }
        }
    }
    return 0;
}