求区间(L, R)小于k的数有多少个

二位数点典中点。
题目链接:数数
以前写的博客链接:
小沙的remake
那题刚好询问的是小于等于a[i]的数有多少个,所以方法二也很好做,但是这题询问的是其他的数 H[i],所以如果用方法二的话,还得离散化,十分麻烦。
用方法一就简单多了,将数组a中小于H[i]的数的下标pos都插入树状数组中,直接查询que® - que(l - 1)就是答案。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 5, mod = 1e9 + 7;
int c[N];
int n;
struct node{
    int val, pos;
    bool operator < (const node &k) const {
        if (val != k.val) return val < k.val;
        return pos < k.pos;
    }
} a[N];
struct kk{
    int l, r, val, pos;
    bool operator < (const kk &k) const{
        return val < k.val;
    }
} d[N];
int ans[N];
void add(int i, int x)
{
    while(i <= n){
        c[i] += x;
        i += (i & -i);
    }
}
int que(int i)
{
    int ret = 0;
    while(i){
        ret += c[i];
        i -= (i & -i);
    }
    return ret;
}
signed main() 
{
    int tt;
    cin >> tt;
    while(tt--){
        int q;
        scanf("%lld%lld", &n, &q);
        for (int i = 1; i <= n; i++) {
            scanf("%lld", &a[i].val);
            a[i].pos = i;
        }
        for (int i = 1; i <= q; i++){
            scanf("%lld%lld%lld", &d[i].l, &d[i].r, &d[i].val);
            d[i].pos = i;
        }
        sort(a + 1, a + 1 + n);
        sort(d + 1, d + 1 + q);
        int k = 1;
        for (int i = 1; i <= q; i++){
            while (a[k].val <= d[i].val && k <= n){
                add(a[k].pos, 1);
                k++;
            }
            ans[d[i].pos] = que(d[i].r) - que(d[i].l - 1);
        }
        for (int i = 1; i <= q; i++) printf("%lld ", ans[i]);
        puts("");
        for (int i = 1; i <= n; i++) {
            a[i].val = a[i].pos = d[i].val = 0;
            c[i] = 0;
        }
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值