uva11610 树状数组+素数打表求因子,好题!

时间:2022-08-12 00:11:30
/*
uva11610
树状数组+素数打表+离散化
打出的素数范围在2-1000000之间,不超过六位数,然后按照格式翻转成七位数
*/
#include<bits/stdc++.h>
using namespace std;
#define maxn 1000005
#define ll long long
int flag[maxn],prime[maxn],cnt;
int fac[maxn],a[maxn],tot,p[maxn];
ll bit1[maxn],bit2[maxn];
map<int,int>m;
int solve(int num){//将每个素数翻转后返回,最后一位必定是0,忽略
int len=,ret=,bit[];
while(num){
bit[len++]=num%;
num/=;
}
for(int i=;i<len;i++)
ret=ret*+bit[i];
while(ret<) ret*=;
return ret;
}
void init(){
for(int i=;i<;i++){//线性筛
if(flag[i])continue;
prime[++cnt]=i;
for(int j=;j*i<;j++)
flag[i*j]=;
}
for(int i=;i<=cnt;i++)//将每个素数翻转
a[i]=solve(prime[i]);
sort(a+,a++cnt);
for(int i=;i<=cnt;i++)//离散化
m[a[i]]=i;
for(int i=;i<=cnt;i++){
fac[i]=;//删掉最后一个0带来的两个质因数
int tmp=a[i];
for(int j=;j<=cnt && prime[j]*prime[j]<=tmp;j++)
while(tmp%prime[j]==){
tmp/=prime[j];
fac[i]++;
}
if(tmp>) fac[i]++;
}
}
void add1(int x,int num){
for(int i=x;i<=maxn-;i+=i&-i)
bit1[i]+=num;
}
void add2(int x,int num){
for(int i=x;i<=maxn-;i+=i&-i)
bit2[i]+=num;
}
ll query1(int x){
ll res=;
for(int i=x;i;i-=i&-i)
res+=bit1[i];
return res;
}
ll query2(int x){
ll res=;
for(int i=x;i;i-=i&-i)
res+=bit2[i];
return res;
}
int main(){
init();
for(int i=;i<=cnt;i++){
add1(i,);
add2(i,fac[i]);
}
char op[];int k;
while(scanf("%s%d",op,&k)!=EOF){
if(op[]=='d'){//把整个点从数组中删除
add1(m[k/],-);
add2(m[k/],-fac[m[k/]]);
}
else {
k++;
int l=,r=cnt,ans;
while(l<=r){
int mid=l+r>>;
ll tmp=query1(mid);
if(tmp<=k)
ans=mid,l=mid+;
else
r=mid-;
}
printf("%lld\n",query2(ans));
}
}
return ;
}