查询区间内等于x的数的个数(分块)

问题:有一个有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;
}

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值