AC日记——线段树练习4 codevs 4919

4919 线段树练习4

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 黄金 Gold

题解

 查看运行结果

 

 

题目描述 Description

给你N个数，有两种操作

1：给区间[a,b]内的所有数都增加X

2：询问区间[a,b]能被7整除的个数

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是add，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是count，表示统计区间[a,b]能被7整除的个数

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3 
2 3 4
6
count 1 3
count 1 2
add 1 3 2
count 1 3
add 1 3 3
count 1 3

样例输出 Sample Output

0

0

0

1

数据范围及提示 Data Size & Hint

10%:1<N<=10,1<Q<=10

30%:1<N<=10000,1<Q<=10000

100%:1<N<=100000,1<Q<=100000

思路：

　　对当前dis取模7；

　　然后根据余数来建立个数；

　　然后每次增加都交换即可；

来，上代码：

#include <cstdio>
#include <iostream>

using namespace std;

struct TreeNodeType {
    int l,r,mid,flag,dis[];
};
struct TreeNodeType tree[<<];

int if_z,n,m;

char Cget;

inline void in(int &now)
{
    now=,if_z=,Cget=getchar();
    while(Cget>''||Cget<'')
    {
        if(Cget=='-') if_z=-;
        Cget=getchar();
    }
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
    now*=if_z;
}

inline void tree_up(int now)
{
    for(int i=;i<=;i++)
    {
        tree[now].dis[i]=tree[now<<].dis[i]+tree[now<<|].dis[i];
    }
}

void tree_build(int now,int l,int r)
{
    tree[now].l=l,tree[now].r=r;
    if(l==r)
    {
        int pos;
        in(pos);
        tree[now].dis[pos%]++;
        return ;
    }
    tree[now].mid=(l+r)>>;
    tree_build(now<<,l,tree[now].mid);
    tree_build(now<<|,tree[now].mid+,r);
    tree_up(now);
}

inline void tree_down(int now)
{
    int pos=tree[now].flag,tmp[];
    tree[now<<].flag+=pos;
    tree[now<<|].flag+=pos;
    for(int i=;i<=;i++) tmp[i]=tree[now<<].dis[i];
    for(int i=;i<=;i++)
    {
        tree[now<<].dis[(i+pos)%]=tmp[i];
        tmp[i]=tree[now<<|].dis[i];
    }
    for(int i=;i<=;i++) tree[now<<|].dis[(i+pos)%]=tmp[i];
    tree[now].flag=;
}

void tree_change(int now,int l,int r,int x)
{
    if(tree[now].l==l&&tree[now].r==r)
    {
        int tmp[];
        for(int i=;i<=;i++) tmp[i]=tree[now].dis[i];
        for(int i=;i<=;i++) tree[now].dis[(i+x)%]=tmp[i];
        tree[now].flag+=x;
        return ;
    }
    if(tree[now].flag) tree_down(now);
    if(l>tree[now].mid) tree_change(now<<|,l,r,x);
    else if(r<=tree[now].mid) tree_change(now<<,l,r,x);
    else
    {
        tree_change(now<<,l,tree[now].mid,x);
        tree_change(now<<|,tree[now].mid+,r,x);
    }
    tree_up(now);
}

int tree_query(int now,int l,int r)
{
    if(tree[now].l==l&&tree[now].r==r) return tree[now].dis[];
    if(tree[now].flag) tree_down(now);
    if(l>tree[now].mid) return tree_query(now<<|,l,r);
    else if(r<=tree[now].mid) return tree_query(now<<,l,r);
    else return tree_query(now<<,l,tree[now].mid)+tree_query(now<<|,tree[now].mid+,r);
}

int main()
{
    in(n);
    tree_build(,,n);
    in(m);char ch[];
    int l,r,x;
    while(m--)
    {
        cin>>ch;
        if(ch[]=='a')
        {
            in(l),in(r),in(x);
            tree_change(,l,r,x);
        }
        else
        {
            in(l),in(r);
            printf("%d\n",tree_query(,l,r));
        }
    }
    return ;
}