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

codevs 4919 线段树练习4

2024-10-12 09:55:02   214人阅读

题目描述 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

//每个节点用一个数组来维护  %7的 = 0 ，1 , 2 , 3 , 4 , 5 , 6 的个数
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>

using namespace std;
const int N=100001;

int n,x,y,z,ans,Q;
int  tmp[7];
string s;

struct node
{
    int l,r;
    int f;
    int sum[7];
}T[N*4]; 

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

void up(int jd)
{
    for(int i=0;i<7;i++) T[jd].sum[i]=T[jd<<1].sum[i]+T[jd<<1|1].sum[i];
}

void build(int jd,int l,int r)
{
    T[jd].l=l; T[jd].r=r;
    if(l==r) 
    {
        scanf("%d",&x);
        x=read();
        T[jd].sum[x%7]++;
        return ;
    }
    int mid=l+r>>1;
    build(jd<<1,l,mid);
    build(jd<<1|1,mid+1,r);
    up(jd);
}

void down(int jd)
{
    for(int i=0;i<7;i++) tmp[i]=T[jd<<1].sum[i];
    for(int i=0;i<7;i++) T[jd<<1].sum[(i+T[jd].f)%7]=tmp[i];
    for(int i=0;i<7;i++) tmp[i]=T[jd<<1|1].sum[i];
    for(int i=0;i<7;i++) T[jd<<1|1].sum[(i+T[jd].f)%7]=tmp[i];
    T[jd<<1].f+=T[jd].f;
    T[jd<<1|1].f+=T[jd].f;
    T[jd].f=0;
}

void add(int jd)
{
    if(T[jd].l>=x&&T[jd].r<=y)
    {
        for(int i=0;i<7;i++) tmp[i]=T[jd].sum[i];
        for(int i=0;i<7;i++) T[jd].sum[(i+z)%7]=tmp[i];
        T[jd].f+=z;
        return;
    }
    if(T[jd].f) down(jd);
    int mid=T[jd].l+T[jd].r>>1;
    if(x<=mid) add(jd<<1);
    if(y>mid) add(jd<<1|1);
    up(jd);
} 

void query(int jd)
{
    if(T[jd].l>=x&&T[jd].r<=y)
    {
        ans+=T[jd].sum[0];
        return;
    }
    if(T[jd].f) down(jd);
    int mid=T[jd].l+T[jd].r>>1;
    if(x<=mid) query(jd<<1);
    if(y>mid) query(jd<<1|1);
}

int main()
{
    n=read();
    build(1,1,n);
    Q=read();
    while(Q--)
    {
        cin>>s;
        if(s[0]==‘a‘)
        {
            x=read();y=read();z=read();
            z%=7;
            add(1);
        }
        else
        {
            x=read();y=read();
            ans=0;
            query(1);
            printf("%d\n",ans);
        }
    }
    return 0;
}

 

codevs 4919 线段树练习4


联系
我们