VI.[Ynoi2016] 掉进兔子洞
bitset
优化莫队。
具体而言,我们似乎只需要求出三个区间的 bitset
,然后求交即可。
但是问题来了,bitset
似乎只支持不可重集求交呀?
没关系,我们考虑在离散化时,对于一个出现次数为 \(c_x\) 的元素 \(x\),预留它后面的 \(c_x\) 个位置(这样总标号数仍为 \(n\))。这样,当将一个数插入区间时,若这是这个数在区间中的第 \(k\) 次出现,就在 bitset
中将第 \(x+k\) 个位置设为 true
,这样便支持了可重集求交。
但是问题又来了,开 \(10^5\) 个 \(10^5\) 的 bitset
吃不消罢?
没关系,我们每 \(\dfrac{10^5}{3}\) 次询问在一起做莫队,这样只需要 \(3.3\times10^9\) 个 bit
,可以过去。
代码:
#include<bits/stdc++.h>
using namespace std;
const int LIM=33334;
const int BBB=320;
int n,m,a[100100],buc[100100];
vector<int>v;
bitset<100010>bs[33350],sb;
int res[33350];
struct node{
int l,r,id;
friend bool operator<(const node&x,const node&y){
if(x.l/BBB!=y.l/BBB)return x.l<y.l;
return (x.l/BBB)&1?x.r>y.r:x.r<y.r;
}
void read(int i){scanf("%d%d",&l,&r),id=i,res[i]+=r-l+1;}
}q[100010];
void Push(int x){sb.set(a[x]+buc[a[x]]++);}
void Pop(int x){sb.reset(a[x]+--buc[a[x]]);}
void solve(){
int M=min(m,LIM);m-=M;
for(int i=1;i<=M;i++){res[i]=0,bs[i].set();for(int j=0;j<3;j++)q[j*M+i].read(i);}
memset(buc,0,sizeof(buc)),sb.reset();
sort(q+1,q+3*M+1);
int L=1,R=0;
for(int i=1;i<=3*M;i++){
while(L>q[i].l)Push(--L);
while(R<q[i].r)Push(++R);
while(L<q[i].l)Pop(L++);
while(R>q[i].r)Pop(R--);
// printf("[%d,%d]\n",q[i].l,q[i].r),cout<<sb<<endl;
bs[q[i].id]&=sb;
}
for(int i=1;i<=M;i++)printf("%d\n",res[i]-3*bs[i].count());
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&a[i]),v.push_back(a[i]);
sort(v.begin(),v.end());
for(int i=1;i<=n;i++)a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin();
// for(int i=1;i<=n;i++)printf("%d ",a[i]);puts("");
solve(),solve(),solve();
return 0;
}