P4688-[Ynoi2016]掉进兔子洞【莫队,bitset】
正题
题目链接:https://www.luogu.com.cn/problem/P4688
题目大意
给出一个长度为\(n\)的序列\(a\)。
然后\(m\)次询问给出三个区间,求这三个区间构成的可重集删去交集后剩下的数字个数和。
\(1\leq n,m\leq 10^5,1\leq a_i\leq 10^9\)
解题思路
考虑用总个数减去重复的个数。
那么现在问题是怎么求三个区间的交集。
假设不考虑重复的数的情况下我们可以莫队处理出三个区间的\(bitset\)然后或起来。
但是现在需要考虑重复的怎么处理,其实也很简单,因为我们是莫队,求出每个数字在序列中小于它的数字个数\(x\),如果这个数字是第\(i\)次出现就标号为\(x+i\)就好了,这样就不会重复了。
然后会发现空间不够大,直接把所有询问分成若干小组来处理就好了。
时间复杂度:\(O(\frac{nm}{\omega}+m\sqrt n)\)
话说bitset的count竟然是\(O(\frac{n}{\omega})\)的/xia。
code
#include
#include
#include
#include
#include
using namespace std;
const int N=1e5+10,M=2e4+10;
struct node{
int l,r,t;
}q[N];
int n,m,k,T,d[N],b[N],c[N],ans[M];
bool v[M];bitsetbst[M],now;
bool cmp(node x,node y)
{return (x.l/T==y.l/T)?(x.rq[i].r)Del(b[r]),r--;
while(lq[i].l)l--,Add(b[l]);
if(!v[q[i].t])
bst[q[i].t]=now,v[q[i].t]=1;
else bst[q[i].t]&=now;
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]-bst[i].count()*3);
return;
}
int main()
{
// freopen("xp1.in","r",stdin);
// freopen("xp1.out","w",stdout);
scanf("%d%d",&n,&m);
T=sqrt(n);
for(int i=1;i<=n;i++)
scanf("%d",&c[i]),d[i]=c[i];
sort(d+1,d+1+n);
for(int i=1;i<=n;i++){
int l=1,r=n;
while(l<=r){
int mid=(l+r)>>1;
if(d[mid]>=c[i])r=mid-1;
else l=mid+1;
}
b[i]=l;
}
for(int i=0;i