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;

}

 

posted @ 2012-03-22 15:14  yejinru  阅读(280)  评论(0编辑  收藏  举报