树桩数组维护区间不同值的个数

题目

原本想用莫队水一水

结果只有80分，不够学了一招。

将问题转化为离线问题。

我们先将询问按右节点升序排序。

然后对于每一个值，我们可以肯定的是，只有最靠近询问右端的值才可能起作用。

case 1 ：如果询问不包括某个值。那么这个值，对答案没有影响

case 2 ：包括的话，那么至少包括1个这样的值，而我们询问的是不同的个数，所以对答案也没有影响

这样的话，我们只需要维护最靠右的位置在哪就行了。

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
struct BIT
{
    int num;
    int data[1000100];
    void insert(int value,int pos)
    {
        while(pos<=num)
        {
            data[pos]+=value;
            pos+=(pos&(-pos));
        }
    }
    int sum(int pos)
    {
        int ans=0;
        while(pos)
        {
            ans+=data[pos];
            pos-=(pos&(-pos));
        }
        return ans;
    }
    int check(int l,int r)
    {
        return sum(r)-sum(l-1);
    }
};
BIT bit;
int last[1000100];
int base[500100];
int ans[200100];
struct node
{
    int left;
    int right;
    int num;
};
bool compare(const node &a,const node &b)
{
    if(a.right==b.right)
        return a.left<b.left;
    return a.right<b.right;
}
int read()
{
    int s=0,f=1;
    char in=getchar();
    while(in<'0'||in>'9')
    {
        if(in=='-')
            f=-1;
        in=getchar();
    }
    while(in>='0'&&in<='9')
    {
        s=(s<<1)+(s<<3)+in-'0';
        in=getchar();
    }
    return s*f;
}
node query[200100];
int main()
{
    int n=read();
    for(int i=1;i<=n;i++)
        base[i]=read();
    bit.num=n;
    int m=read();
    for(int i=1;i<=m;i++)
    {
        query[i].left=read();
        query[i].right=read();
        query[i].num=i;//由于是离线操作吗——
    }
    sort(query+1,query+1+m,compare);//排序，左端点其实可以不拍
    int i=0;//i是当前遍历到的右端点
    for(int j=1;j<=m;j++)
    {
        while(i<query[j].right)
        {
            i+=1;
            if(last[base[i]])
            {
                bit.insert(-1,last[base[i]]);
                bit.insert(1,i);
                last[base[i]]=i;
            }
            else
            {
                bit.insert(1,i);
                last[base[i]]=i;
            }
        }
        //printf("%d\n",bit.check(query[i].left,query[i].right));
        ans[query[j].num]=bit.check(query[j].left,query[j].right);
    }
    for(i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}

转载于:https://www.cnblogs.com/Lance1ot/p/8921788.html