经理的烦恼
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");
}
}