首页 > 代码库 > codevs 4919 线段树练习4

codevs 4919 线段树练习4

4919 线段树练习4

 时间限制: 1 s
 空间限制: 128000 KB
 
 
题目描述 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 46count 1 3count 1 2add 1 3 2count 1 3add 1 3 3count 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

 

每个节点维护%7的=0,1,2,3,4,5,6的个数

#include<cstdio>#define N 100001using namespace std;int n,x,y,z,ans;int  tmp[7];struct node{    int l,r;    int f;    int sum[7];}tr[N*4]; void up(int k){    for(int i=0;i<7;i++) tr[k].sum[i]=tr[k<<1].sum[i]+tr[k<<1|1].sum[i];}void build(int k,int l,int r){    tr[k].l=l; tr[k].r=r;    if(l==r)     {        scanf("%d",&x);        tr[k].sum[x%7]++;         return ;    }    int mid=l+r>>1;    build(k<<1,l,mid);    build(k<<1|1,mid+1,r);    up(k);}void down(int k){    for(int i=0;i<7;i++) tmp[i]=tr[k<<1].sum[i];    for(int i=0;i<7;i++) tr[k<<1].sum[(i+tr[k].f)%7]=tmp[i];    for(int i=0;i<7;i++) tmp[i]=tr[k<<1|1].sum[i];    for(int i=0;i<7;i++) tr[k<<1|1].sum[(i+tr[k].f)%7]=tmp[i];    tr[k<<1].f+=tr[k].f;    tr[k<<1|1].f+=tr[k].f;    tr[k].f=0;}void add(int k){    if(tr[k].l>=x&&tr[k].r<=y)    {        for(int i=0;i<7;i++) tmp[i]=tr[k].sum[i];        for(int i=0;i<7;i++) tr[k].sum[(i+z)%7]=tmp[i];        tr[k].f+=z;        return;    }    if(tr[k].f) down(k);    int mid=tr[k].l+tr[k].r>>1;    if(x<=mid) add(k<<1);    if(y>mid) add(k<<1|1);    up(k);} void query(int k){    if(tr[k].l>=x&&tr[k].r<=y)    {        ans+=tr[k].sum[0];        return;    }    if(tr[k].f) down(k);    int mid=tr[k].l+tr[k].r>>1;    if(x<=mid) query(k<<1);    if(y>mid) query(k<<1|1);}int main(){    scanf("%d",&n);    build(1,1,n);    scanf("%d",&n);    char ch[8];    while(n--)    {        scanf("%s",ch);        if(ch[0]==a)        {            scanf("%d%d%d",&x,&y,&z);            z%=7;            add(1);        }        else        {            scanf("%d%d",&x,&y);            ans=0;            query(1);            printf("%d\n",ans);        }    }} 

 

codevs 4919 线段树练习4