对每个询问,单独输出一行,表示sl...sr中权值∈[a,b]的权值的种类数。
如果用莫队处理询问,能用树状数组做到O(m*sqrt(n)*logn)
不过这样显然是T飞的。。。用个小技巧
对于权值分块维护,这样单次查询能O(sqrt(n)),修改O(1)
总复杂度就能降一个log了
#include<iostream> #include<cstdio> #include<algorithm> #include<cmath> #include<cstring> #include<vector> #include<queue> #include<set> #include<map> #include<stack> #include<bitset> #include<ext/pb_ds/priority_queue.hpp> using namespace std; const int maxn = 1E5 + 10; const int maxm = 1E6 + 10; int n,m,Sqrt,sum[320],cnt[maxn],ans[maxm],s[maxn]; int Getpos(int x) { return (x % Sqrt == 0)?x / Sqrt:x / Sqrt + 1; } struct Query{ int l,r,a,b,num; Query(){} Query(int l,int r,int a,int b,int num): l(l),r(r),a(a),b(b),num(num){} bool operator < (const Query &B) const { if (Getpos(l) < Getpos(B.l)) return 1; if (Getpos(l) > Getpos(B.l)) return 0; return r < B.r; } }Q[maxm]; void Add(int x) { if (!cnt[x]) ++sum[Getpos(x)]; ++cnt[x]; } void Dec(int x) { if (cnt[x] == 1) --sum[Getpos(x)]; --cnt[x]; } int getint() { char ch = getchar(); int ret = 0; while (ch < '0' || '9' < ch) ch = getchar(); while ('0' <= ch && ch <= '9') ret = ret*10 + ch - '0',ch = getchar(); return ret; } int main() { #ifdef DMC freopen("DMC.txt","r",stdin); #endif n = getint(); m = getint(); Sqrt = sqrt(n); for (int i = 1; i <= n; i++) s[i] = getint(); for (int i = 1; i <= m; i++) { int l,r,a,b; l = getint(); r = getint(); a = getint(); b = getint(); Q[i] = Query(l,r,a,b,i); } sort(Q + 1,Q + m + 1); int L = 1,R = 0; for (int i = 1; i <= m; i++) { while (R < Q[i].r) Add(s[++R]); while (R > Q[i].r) Dec(s[R--]); while (L < Q[i].l) Dec(s[L++]); while (L > Q[i].l) Add(s[--L]); int Ans = 0,pa = Getpos(Q[i].a),pb = Getpos(Q[i].b); if (pa == pb) { for (int j = Q[i].a; j <= Q[i].b; j++) Ans += (cnt[j])?1:0; } else { for (int j = pa + 1; j < pb; j++) Ans += sum[j]; for (int j = Q[i].a,t = pa*Sqrt; j <= t; j++) Ans += (cnt[j])?1:0; for (int j = pb*Sqrt - Sqrt + 1; j <= Q[i].b; j++) Ans += (cnt[j])?1:0; } ans[Q[i].num] = Ans; } for (int i = 1; i <= m; i++) printf("%d\n",ans[i]); return 0; }