HOJ 1867 经理的烦恼 树状数组

/*

题目：

给出n个商店，每个商店有初始化的商品数量，现在有两种指令：

0 x y 连锁店x的商品数量变化值为y，y > 0商品数量增加， y < 0减少

1 i j 输出编号在[i,j]区间内的连锁店中商品数量为素数的有多少家

当指令为1时，求给出 i到j的商品数量为素数的商店的数目

分析：

用树状数组做，只不过在update()处改动一下，并且注意到有可能改动完后和改动前

均为素数的情况，所以用visit[]数组记录此前是否为素数，另外需要注意的是当初始

化时即为奇数的话，当时就要初始化数组c[]以及visit[]数组

*/

#include <iostream>

#include <cstdio>

#include <cstring>

using namespace std;

const int X = 1000010;

int a[X],c[X],n,m,k;

bool visit[X];

bool check(int x) //判断是否是素数

{

if(x<=1) //小于等于一直接返回false

return false;

for(int i=2;i*i<=x;i++)

if(x%i==0)

return false;

return true;

}

int lowbit(int x)

{

return x & -x;

}

int sum(int x) //求和

{

int ans = 0;

while(x>0)

{

ans += c[x];

x -= lowbit(x);

}

return ans;

}

void update(int x)

{

if(check(a[x])) //如果这个商店的商品数为素数

{

if(!visit[x]) //如果之前不是素数的话，现在为素数

{

visit[x] = true;

while(x<=n)

{

c[x]++;

x += lowbit(x);

}

}

}

else //如果这个商店的商品数为偶数

{

if(visit[x]) //如果之前是素数，但现在不是素数

{

visit[x] = false;

while(x<=n)

{

c[x]--;

x += lowbit(x);

}

}

}

}

int main()

{

freopen("sum.in","r",stdin);

freopen("sum.out","w",stdout);

int x,y,z,cnt = 0;

while(scanf("%d%d%d",&n,&m,&k),n||m||k)

{

printf("CASE #%d:\n",++cnt);

for(int i=1;i<=n;i++) //初始化

a[i] = k;

if(check(k)) //如果刚开始即为素数

{

for(int i=1;i<=n;i++)

{

c[i] = lowbit(i);

visit[i] = true; //visit数组为true时表示之前的数为素数

}

}

else

{

memset(c,0,sizeof(c));

memset(visit,false,sizeof(visit));

}

for(int i=1;i<=m;i++)

{

scanf("%d%d%d",&z,&x,&y);

if(z)

printf("%d\n",sum(y)-sum(x-1));

//计算的是[i,j]的区间内的连锁店中商品数量为素数的有多少家

else

{

a[x] += y; //x商店的变化

update(x); //更新x商店的变化给其他商店带来的变化

}

}

cout<<endl;

}

return 0;

}