[Ynoi2016] 掉进兔子洞

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;
}