HIT 1867 经理的烦恼

点击打开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;
}