hoj1867 经理的烦恼（树状数组）

经理的烦恼

Jerry是一家公司销售部门的经理。这家公司有很多连锁店，编号为1，2，3，… Jerry每天必须关注每家连锁店的商品数量及其变化，一项很乏味的工作。在连锁店比较少的时候，Jerry喜欢计算编号在[i,j]区间内的连锁店中商品数量为素数的有多少家，但是现在连锁店的数量急剧增长，计算量很大，Jerry很难得出结果。
输入格式
题目有多组输入。每组输入第一行有三个整数：C 连锁店的数量 N 指令的条数 M 每家连锁店初始的商品数量
接下来有N行，每行有一条指令。指令的格式为：
0 x y 连锁店x的商品数量变化值为y，y > 0商品数量增加， y < 0减少
1 i j 输出编号在[i,j]区间内的连锁店中商品数量为素数的有多少家
1 <= i, x, j < 1000000 连锁店中的商品数量a满足 0 <= a < 10000000，C = N = M = 0标志输入结束
输出格式
对于每组输入，输出它的序号。对于一组输入中的1指令输出要求的整数。每组输出后打印一行空行。

分析：树状数组记录区间素数个数，改变时判断一下连锁店x是否改变状态。

代码

#include <cstdio>
#include <cstring>
#include <cmath>
#define maxn 2000000
using namespace std;

int a[maxn],c[maxn],n,m,w,p,x,y;
bool f[11000006];

int insert(int x,int y)
{
        while (x<=n)
        {
                c[x]+=y;
                x+=x&(-x);
        }
        return 0;
}

int count(int x)
{
        int s=0;
        while (x>0)
        {
                s+=c[x];
                x-=x&(-x);
        }
        return s;
}

int check(int x)
{
    if (x==0||x==1) return false; 
    for (int i=2;i<=sqrt(x);i++)  
    {  
        if(x%i==0)  
            return false;  
    }  
    return true; 
}

int main()
{   
        int q=0;
        while(scanf("%d%d%d",&n,&m,&w), n != 0 || m != 0 || w != 0)
        {
                q++;
                printf("CASE #%d:\n",q);
                for (int i=1;i<=n;i++)
                        a[i]=w;
                memset(c,0,sizeof(c));
                if (check(w)) 
                {
                        for (int i=1;i<=n;i++)
                                insert(i,1);
                }
                for (int i=1;i<=m;i++)
                {
                        scanf("%d%d%d",&p,&x,&y);
                        if (p==0)
                        {
                                bool f1=check(a[x]),f2=check(a[x]+y);
                                if (f1&&!f2) insert(x,-1);
                                else if (!f1&&f2) insert(x,1);
                                a[x]+=y;
                        }
                        else
                        {
                                printf("%d\n",count(y)-count(x-1));                    
                        }
                }
                printf("\n");
        }
}