洛谷 – 求区间不同数字的个数(树状数组 + 思维)

Aurora141592

于 2020-12-30 13:03:18 发布

阅读量1.4k

点赞数 2

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/wulalalawu/article/details/111972055

版权

本文介绍了如何利用树状数组在遇到区间内不同数字个数的问题时，避免莫队算法的TLE情况。通过将区间右端点排序，并仅考虑每个数字的最后一次出现位置，可以有效计算区间内不同数字的个数。具体做法是对数组进行遍历，更新树状数组，将当前位置设为1，若数字之前出现过，则将之前位置设为0，最后求区间和即为答案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://www.luogu.org/problemnew/show/P1972
莫队TLE了，于是写了个树状数组。
按右端点排序，当右端点固定下来之后，右端点之前的每个数字只用考虑最后一次出现的位置。这是因为，假设右端点之前有某个数字出现了两次，那么当左端点在第一个位置之前和两个位置之间对答案的贡献都是1，在第二个位置之后对答案贡献是0，所以每次只用考虑最后一个出现的位置。那么这道题就可以用树状数组解决，遍历一遍数组，每次都把当前位置变成1，如果这个数字在之前出现过，就把之前出现的位置变成0，最后求区间和就是答案。

#include<bits/stdc++.h>
using namespace std;
int bit[500005], a[500005], p[1000005], ans[200005], n, m;
int query(int x)
{
   
    int sum = 0;
    while(x){
   
        sum += bit[x];
        x -= x & -x

最低0.47元/天解锁文章