ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

【算法模板】离线树状数组(区间查询小于等于x的数个数)

2022-04-18 21:33:56  阅读:168  来源: 互联网

标签:struct 树状 int 离线 pos ans query BIT 模板


只需要把询问按x升序排序,在查询的过程中不断让树状数组把<=x元素的下标处+1即可。(为此,把序列按val排序)

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;

pair<int, int> a[N];
#define val first
#define pos second

struct query {
    int l, r, x, pos, ans;
} q[N];

int n, m;

struct BIT {
    int t[N];
    BIT() { memset(t, 0, sizeof(t)); }
    void add(int i, int x) {
        for (; i <= n; i += i & -i)
            t[i] += x;
    }
    int sum(int i) {
        int res = 0;
        for (; i; i -= i & -i)
            res += t[i];
        return res;
    }
};

int main() {
    ios::sync_with_stdio(false); 
    cin.tie(0);
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i].val;
        a[i].pos = i;
    }
    for (int i = 1; i <= m; i++) {
        cin >> q[i].l >> q[i].r >> q[i].x;
        q[i].pos = i; q[i].ans = 0;
    }
    sort(a + 1, a + 1 + n);
    sort(q + 1, q + 1 + m, [&](query a, query b) { return a.x < b.x; });
    
    BIT T; int p = 0;
    for (int i = 1; i <= m; i++) {
        query& Q = q[i];
        while (a[p + 1].val <= Q.x && p + 1 <= n) {
            p++;
            T.add(a[p].pos, 1);
        }
        Q.ans = T.sum(Q.r) - T.sum(Q.l - 1);
    }
    sort(q + 1, q + 1 + m, [&](query a, query b) { return a.pos < b.pos; });
    for (int i = 1; i <= m; i++)
        cout << q[i].ans << "\n";
    return 0;
}

标签:struct,树状,int,离线,pos,ans,query,BIT,模板
来源: https://www.cnblogs.com/vv123/p/16162872.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有