[Codevs] 5037 线段树练习4加强版

5037 线段树练习4加强版

 时间限制: 4 s

 空间限制: 256000 KB

 题目等级 : 钻石 Diamond

 

 

题目描述 Description

维护一个序列，要求支持下列2种操作：

add a b c:区间[a,b]中每个数加上c

count a b:查询区间[a,b]中有多少数是k的倍数（k为给定常数）

 

输入描述 Input Description

第一行三个数n,m,k,分别表示序列长度、操作数和count中的k

接下来一行n个整数，表示原始序列

接下来m行，每行是题面中的操作之一

 

输出描述 Output Description

对于每个count操作，输出一行答案

 

样例输入 Sample Input

10 10 5

5 5 8 3 5 6 7 8 3 0

add 2 7 1

count 3 4

add 2 5 4

count 1 5

count 2 6

count 1 3

add 4 8 3

count 3 7

add 4 8 2

count 1 2

 

样例输出 Sample Output

0

3

2

2

1

2

 

数据范围及提示 Data Size & Hint

10%:n,m<=10,k<=10000;

另外的20%:n,m<=100000,k<=10;

另外的20%:n,m<=50000,k<=100;

100%:n,m<=200000,k<=200000.

 

分析 Analysis

分块！

拿线段树练习4 的代码改了改就过了

模型和线段树练习4是一样的，只不过改了取模数的范围

 

代码 Code

#include<cstdio>
#include<iostream>
#include<cmath>
#define maxn 1000000
using namespace std;

char str[10];
int block[maxn],k,mod[450][200010],size,arr[maxn],add[450],bdd[450],n,m;

void modify(){
    int a,b,c;
    scanf("%d%d%d",&a,&b,&c);
    a--,b--,c %= k;
    if(!c) return;
    
    if(block[a] == block[b]){
        for(int i = a;i <= b;i++){
            mod[block[i]][arr[i]]--;
            arr[i] = (arr[i]+c)%k;
            mod[block[i]][arr[i]]++;
        }
    }else{
        
        for(int i = a;i <= block[a]*size-1;i++)
            mod[block[i]][arr[i]]--,
            arr[i] = (arr[i]+c)%k,
            mod[block[i]][arr[i]]++;
        for(int i = (block[b]-1)*size;i <= b;i++)
            mod[block[i]][arr[i]]--,
            arr[i] = (arr[i]+c)%k,
            mod[block[i]][arr[i]]++;
        for(int i = block[a]+1;i < block[b];i++)
            add[i] = (add[i]+c)%k,
            bdd[i] = (bdd[i]-c+k)%k;
            
    }
}

void count(){
    int a,b;
    scanf("%d%d",&a,&b);
    a--,b--;
    int ans = 0;
    
    if(block[a] == block[b]){
        for(int i = a;i <= b;i++)
            ans += ((arr[i]+add[block[i]])%k)?0:1;
    }else{
        
        for(int i = a;i <= block[a]*size-1;i++)
            ans += ((arr[i]+add[block[i]])%k)?0:1;
        for(int i = (block[b]-1)*size;i <= b;i++)
            ans += ((arr[i]+add[block[i]])%k)?0:1;
        for(int i = block[a]+1;i < block[b];i++)
            ans += mod[i][bdd[i]];
        
    }
    
    printf("%d\n",ans);
}

void show(){
    for(int i = 0;i < n;i++){
        printf("%d ",arr[i]+add[block[i]]);
    }cout << endl;
}

int main(){
    scanf("%d%d%d",&n,&m,&k);
    size = (int)sqrt(n)+1;
    for(int i = 0;i < n;i++) block[i] = i/size+1;
//    for(int i = 0;i < n;i++) printf("#%d: block %d\n",i,block[i]);
    for(int i = 0;i < n;i++){
        scanf("%d",&arr[i]);
        arr[i] %= k;
        mod[block[i]][arr[i]]++;
    }
//    show();
    for(int i = 0;i < m;i++){
        scanf("%s",str);
        if(str[0] == 'a') modify();
        else count();
//        show();
    }
    
    return 0;
}