CodeVs——T 4919 线段树练习4

http://codevs.cn/problem/4919/

 时间限制: 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

#include <algorithm>
#include <cstdio>

using namespace std;

const int N(100000+5);
int n,x,y,z,q;

#define lc (now<<1)
#define rc (now<<1|1)
#define mid (l+r>>1)
struct AC_Tree
{
    int val[7],flag;
}tree[N<<2];
inline void pushup(int now)
{
    for(int i=0;i<7;i++)
        tree[now].val[i]=tree[lc].val[i]+tree[rc].val[i];
}
inline void pushdown(int now)
{
    int temp[7],flag=tree[now].flag; tree[now].flag=0;
    tree[lc].flag+=flag; tree[rc].flag+=flag;
    for(int i=0;i<7;i++) temp[i]=tree[lc].val[i];
    for(int i=0;i<7;i++) tree[lc].val[((i+flag)%7)]=temp[i];
    for(int i=0;i<7;i++) temp[i]=tree[rc].val[i];
    for(int i=0;i<7;i++) tree[rc].val[((i+flag)%7)]=temp[i];
}
void Build(int now,int l,int r)
{
    if(l==r)
    {
        scanf("%d",&x);
        tree[now].val[x%7]++;
        return ;
    }
    Build(lc,l,mid);
    Build(rc,mid+1,r);
    pushup(now);
}
void Change(int now,int l,int r,int L,int R,int x)
{
    if(l==L&&r==R)
    {
        int temp[7];
        tree[now].flag+=x;
        for(int i=0;i<7;i++) temp[i]=tree[now].val[i];
        for(int i=0;i<7;i++) tree[now].val[(i+x)%7]=temp[i];
        return ;
    }
    if(tree[now].flag) pushdown(now);
    if(R<=mid) Change(lc,l,mid,L,R,x);
    else if(L>mid) Change(rc,mid+1,r,L,R,x);
    else Change(lc,l,mid,L,mid,x),Change(rc,mid+1,r,mid+1,R,x);
    pushup(now);
}
int Query(int now,int l,int r,int L,int R)
{
    if(l==L&&r==R) return tree[now].val[0];
    if(tree[now].flag) pushdown(now);
    if(R<=mid) return Query(lc,l,mid,L,R);
    else if(L>mid) return Query(rc,mid+1,r,L,R);
    else return Query(lc,l,mid,L,mid)+Query(rc,mid+1,r,mid+1,R);
}

int main()
{
    scanf("%d",&n);
    Build(1,1,n);
    scanf("%d",&q);
    for(char s[5];q--;)
    {
        scanf("%s%d%d",s,&x,&y);
        if(s[0]=='a')
        {
            scanf("%d",&z);
            Change(1,1,n,x,y,z);
        }
        else printf("%d
",Query(1,1,n,x,y));
    }
    return 0;
}