问题:有一个有n个数的数组,q次查询,每次查询l, r ,x,查询区间[l,r]内等于x的数的个数
思路:分块。
就把这题当成是分块的入门题来讲解一下分块。分块其实就是一种比较优美的暴力(我觉得),一般的分块都是把长度为n的数组分成每一块为sqrt(n)个数的多个块。然后对于区间的操作就可以不再是一个一个数进行处理,而是可以一个块一个块的处理,每次查询最多会涉及到sqrt(n)个块,这样复杂度就降了下来。
(先进行几点说明,本文中的数组下标从1开始,块的编号也是从1开始,每个块的大小k = sqrt(n))
分块过后的对于区间[l,r]的操作就分为两种情况:
设cl为区间左端点l所在的块号,cr为区间右端点r所在的块号。那么怎么求一个下标所属的块号呢,很简单:(x-1)/k + 1。
第一种情况是cl == cr,也就是左右端点在同一个块内,这样直接从l循环到r处理操作就可以了,复杂度很显然最多是sqrt(n)
第二种情况我们可以统一的假定为 cl != cr,这样我们可以把区间分为三部分,一是l到cl*k这部分可以直接循环操作,后面就是第cl+1块到cr-1块这部分,最后就是(cr-1)*k+1到r这部分。
这一题要求的是一个区间内等于x的有多少个数,我们可以对每一个块进行排序,然后就可以用二分求出等于x的个数。
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 300010;
int a[maxn];
int b[maxn];
int n,k;
int li(int x) //求x的块号
{
return (x-1)/k + 1;
}
int query(int l,int r,int x)
{
int i;
int cl = li(l);
int cr = li(r);
int cnt = 0;
if(cl == cr)
{
for(i=l;i<=r;i++)
if(a[i] == x)
cnt++;
return cnt;
}
for(i=l;i<=cl*k;i++)
{
if(a[i] == x)
cnt++;
}
for(i=cl+1;i<cr;i++)
{
cnt += upper_bound(b+(i-1)*k+1,b+i*k+1,x) - lower_bound(b+(i-1)*k+1,b+i*k+1,x);
}
for(i=(cr-1)*k+1;i<=r;i++)
{
if(a[i] == x)
cnt++;
}
return cnt;
}
int main(void)
{
int i,j;
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
b[i] = a[i];
}
k = sqrt(n);
for(i=1;i<=k+1;i++)
{
if((i-1)*k+1 > n)
break;
sort(b+(i-1)*k+1,b+min(i*k,n)+1);
}
int q;
scanf("%d",&q);
while(q--)
{
int l,r,x;
scanf("%d%d%d",&l,&r,&x);
int ans = query(l,r,x);
printf("%d\n",ans);
}
return 0;
}