[Codevs] 4919 线段树练习4

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

 

分析 Analysis

分块！

相比线段树分块真的写起来很方便啊qwq

每个块保存块内关于 7 的剩余系的元素数量

（就是%7以后 0~6 的数量啦）

关键是add 操作，真恶心

用以前线段树的那套修改方案，最多过3个点

看了看TJM的方案：

每次整块增值时拉低bdd（负责保存历经修改以后整除 7 的数字到底是哪一个）

查询时计数 mod[ block[ i ] ][ bdd[ i ] ]

 

代码 Code 

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

char str[maxn/100];
int block[maxn],mod[maxn][10],size,arr[maxn],add[maxn],bdd[maxn],n,m;

void modify(){
    int a,b,c;
    scanf("%d%d%d",&a,&b,&c);
    a--,b--,c %= 7;
    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)%7;
            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)%7,
            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)%7,
            mod[block[i]][arr[i]]++;
        for(int i = block[a]+1;i < block[b];i++)
            add[i] = (add[i]+c)%7,
            bdd[i] = (bdd[i]-c+7)%7;
            
    }
}

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]])%7)?0:1;
    }else{
        
        for(int i = a;i <= block[a]*size-1;i++)
            ans += ((arr[i]+add[block[i]])%7)?0:1;
        for(int i = (block[b]-1)*size;i <= b;i++)
            ans += ((arr[i]+add[block[i]])%7)?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",&n);
    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] %= 7;
        mod[block[i]][arr[i]]++;
    }
//    show();
    scanf("%d",&m);
    for(int i = 0;i < m;i++){
        scanf("%s",str);
        if(str[0] == 'a') modify();
        else count();
//        show();
    }
    
    return 0;
}