【ICPC-410】HIT 1867 经理的烦恼

最新推荐文章于 2020-01-24 13:50:12 发布

原创最新推荐文章于 2020-01-24 13:50:12 发布 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

3. ACM-ICPC 专栏收录该内容

467 篇文章

订阅专栏

本文详细介绍了如何运用树状数组解决给定区间内质数数量的问题，包括初始化、更新操作和查询操作的具体实现。通过判断初始值是否为质数进行初始化，利用树状数组高效地更新和查询区间内的质数个数。

点击打开HIT 1867

思路: 树状数组
分析:
1 题目要求的是给定一个区间求这个区间质数的个数
2 题目给定n条命令和每个店的初始的值，那么我们初始化的时候就要通过判断给定的初始值是否为质数来初始化
3 因为要求的是质数的个数，那么我们可以这么想，假设现在改变了店铺x的值，那么我们通过判断前后是否是质数的关系来更新树状数组
4 求区间的质数的个数的时候直接求即可

代码:


#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int MAXN = 1000010;

int n , val;
int num[MAXN];
int treeNum[MAXN];

int lowbit(int x){
    return x&(-x);
}

int getSum(int x){
    int sum = 0;
    while(x){
         sum += treeNum[x];
         x -= lowbit(x);
    }
    return sum;
}

void add(int x , int val){
    while(x < MAXN){
         treeNum[x] += val;
         x += lowbit(x);
    }
}

// 注意判断质数的写法
bool isPrime(int x){
    if(x <= 1)
        return false;
    if(x == 2)
        return true;
    int tmp = sqrt(x);
    for(int i = 2 ; i <= tmp ; i++)
        if(x%i == 0)
            return false;
    return true;
}

void init(){
    int x = 0;
    if(isPrime(val))
        x = 1;
    memset(treeNum , 0 , sizeof(treeNum));
    for(int i = 1 ; i <= n ; i++){
        num[i] = val;
        add(i , x); 
    }
}

void solve(int mark , int x , int y){
    if(mark == 1){
        int ans = getSum(y);    
        ans -= getSum(x-1); 
        printf("%d\n" , ans); 
    }
    else{
        int tmp = num[x];
        num[x] += y;
        if(isPrime(num[x])){
            if(!isPrime(tmp))    
               add(x , 1);
        }
        else{
            if(isPrime(tmp))    
               add(x , -1);
        }
    }
}

int main(){
    int cas = 1; 
    int m , mark;
    int x , y; 
    while(scanf("%d%d%d" , &n , &m , &val) &&n+m+val){
         init(); 
         printf("CASE #%d:\n" , cas++);
         while(m--){
              scanf("%d%d%d" , &mark , &x , &y); 
              solve(mark , x , y);
         }
         puts(""); 
    }
    return 0;
}