BZOJ 2743 采花(树状数组)

时间:2021-08-28 21:11:41

题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=2743

题意:给出一个数列,每个询问查询[L,R]中至少出现两次的数字有多少种?

思路:(1)记录每个位置i的数字的前一个相同数字出现的位置pre[i],没有前一个相同的pre[i]为0。

(2)询问按照R升序。

(3)开始计算:枚举i从1到m(m为询问个数),对于某个位置x,将pre[pre[x]]+1增加1,pre[x]+1减少1,这样做的原因在于任意一个出现两次的数字在任意一段区间中最多只插入一次。然后统计左区间之前的和即可。

int s[N];

void add(int x,int t)
{
    while(x<N)
    {
        s[x]+=t;
        x+=x&-x;
    }
}

int get(int x)
{
    int ans=0;
    while(x)
    {
        ans+=s[x];
        x-=x&-x;
    }
    return ans;
}

int a[N],pre[N],last[N],n;

struct node
{
    int L,R,id;
};

node b[N];
int m,ans[N],K;

int cmp(node a,node b)
{
    return a.R<b.R;
}

int main()
{
    RD(n,K,m);
    int i;
    FOR1(i,n)
    {
        RD(a[i]);
        pre[i]=last[a[i]];
        last[a[i]]=i;
    }
    FOR1(i,m) RD(b[i].L,b[i].R),b[i].id=i;
    sort(b+1,b+m+1,cmp);
    int now=0;
    FOR1(i,m)
    {
        while(now<b[i].R)
        {
            now++;
            add(pre[pre[now]]+1,1);
            add(pre[now]+1,-1);
        }
        ans[b[i].id]=get(b[i].L);
    }
    FOR1(i,m) PR(ans[i]);
}