P1972 [SDOI2009]HH的项链-树状数组-区间不同数查询

  • P1972 [SDOI2009]HH的项链
  • 题意:一个整数N,表示项链的长度。N 个整数,一个整数M,表示 询问的个数。
  • 接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。最终输出区间内不同数目的个数
  • 思路:最初想法:预处理出,到每个位置时已经出现的不同数字的个数,询问按照L排序,然后id从1开始遍历每一段
  • q[i].L,如果经历过程中遇到的数的终止位置为当前遍历到的位置进行树状数组更新这个位置+1,那么L-R的结果就是
  • 预处理的不同数的个数pre[r]—树状数组中维护的已经消失的数,但是出现了bug 比如 序列 1 2 3 4 5 1查询2-3区间,
  • 此时1的个数就出现了问题。
  • 正确思路:直接按照r排序,id从1开始遍历每一段q[i].r 然后遇到的每一个数都进行updata +1,同时判断这个数上一次出现
  • 的位置,如果出现过,那就把那个位置-1,然后更新这个数的最新的更新位置,每次结果就是getsum(r)-getsum(L-1).
  • #include<bits/stdc++.h>
    #define maxn 1234567
    int n,m,x,y,pre[maxn],tot,r;
    int a[maxn],ans[maxn],sum[maxn];
    struct node
    {
        int l,r,id;
        bool operator<(const node &b)const
        {
            return r<b.r;
        }
    } q[maxn];
    int lowbit(int x)
    {
        return x&(-x);
    }
    void updata(int x,int d)
    {
        while(x<=n)
        {
            sum[x]+=d;
            x=x+lowbit(x);
        }
    }
    int getsum(int x)
    {
        int res=0;
        while(x>0)
        {
            res+=sum[x];
            x=x-lowbit(x);
        }
        return res;
    }
    int main()
    {
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        scanf("%d",&m);
        for(int i=0; i<m; i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        std::sort(q,q+m);
        r=1;
        for(int i=0; i<m; i++)
        {
            while(r<=q[i].r)
            {
                if(pre[a[r]]!=0)
                    updata(pre[a[r]],-1);
                pre[a[r]]=r;
                updata(r,1);
                r++;
            }
            ans[q[i].id]=getsum(q[i].r)-getsum(q[i].l-1);
        }
        for(int i=0; i<m; i++)
            printf("%d\n",ans[i]);
        return 0;
    }
    

猜你喜欢

转载自blog.csdn.net/BePosit/article/details/84591430