利用树状数组离线 统计一个区间内不同数字的个数

基本思路是树状数组, 但是有重复的数, 要保证在一个区间内只更新过一次

所做的处理是用先遍历记录数字的数组a, 用数-位置数组 mp[a[i]] 来记录每个数第一次出现的位置并该点更新

这个操作对应的代码

1 for(int i = 1; i <= n; i++) {
2     if(mp[a[i]] == 0) { //如果是第一次出现, 记录第一次出现位置并更新
3         Add(i, 1);
4         mp[a[i]] = i;
5     }
6 }
View Code

记录左右查询, 每个查询按区间右端点R[i]从小到大排序

设一个扫描线k, 它的目的保证R[i]之前所有不同的点只更新过一次, 并且是在离R[i]最近的那个点更新

如此便可以写出这样一段代码

 1 for(int i = 1; i <= query_num; i++) {
 2     while(k <= R[i]) {
 3         if(mp[a[i]] != k) {     //如果不是最新次出现
 4             Add(mp[a[k]], -1);  //将上一次出现的更新-1
 5             Add(k, 1);         //将这个位置新出现的更新1
 6             mp[a[k]] = k;       //更新这个数最近一次的位置
 7         }
 8         k++;
 9     }
10     ans[i] = Sum(ed[i]) - Sum(st[i] - 1);
11 }
View Code

 

转载于:https://www.cnblogs.com/Quinte/p/4898083.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
树状数组(Binary Indexed Tree,BIT)是一种据结构,用于维护序列的前缀和。它可以支持单点修改和区间查询,但是不支持区间修改。不过可以通过巧妙的方式将区间修改转化为单点修改,从而实现区间修改。 区间修改:我们可以将区间修改转化为单点修改,具体方法是将区间修改拆成两次单点修改:先将区间左端点减一,再将区间右端点加一。这样的话,查询时只需要查询右端点的前缀和,再减去左端点的前缀和即可得到区间和。 区间查询:查询区间和可以通过前缀和相减得到。具体方法是查询右端点的前缀和,再减去左端点减一的前缀和即可得到区间和。 具体实现可以参考下面的代码: ```C++ #include <iostream> #include <vector> using namespace std; class BIT { public: BIT(int n) : sums(n + 1) {} void update(int i, int delta) { while (i < sums.size()) { sums[i] += delta; i += lowbit(i); } } int query(int i) { int sum = 0; while (i > 0) { sum += sums[i]; i -= lowbit(i); } return sum; } private: vector<int> sums; int lowbit(int x) { return x & (-x); } }; class RangeBIT { public: RangeBIT(int n) : bit1(n), bit2(n) {} void update(int i, int j, int delta) { bit1.update(i, delta); bit1.update(j + 1, -delta); bit2.update(i, delta * (i - 1)); bit2.update(j + 1, -delta * j); } int query(int i, int j) { return prefix_sum(j) - prefix_sum(i - 1); } private: BIT bit1, bit2; int prefix_sum(int i) { return bit1.query(i) * i - bit2.query(i); } }; ``` 注意,这里的 BIT 是用于维护前缀和的,而 RangeBIT 是用于维护区间和的。区间修改时,bit1 用于维护前缀和,bit2 用于维护前缀和乘以下标的和。查询时,先计算出区间的前缀和,再减去左端点减一的前缀和乘以左端点即可得到区间和。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值