牛客练习赛52 B：Galahad（树状数组维护区间不同元素和(个数)）

【题目】

查询区间和，如果区间元素重复出现则计数一次。

【题解】

按区间的右端点建立树状数组，维护区间[1,R]的每个元素的最右位置。按查询区间的右端点排序，依次处理，每次更新当前值的最右位置即可。

若要查询区间不同元素个数，把

        for(;it<=q[i].R;it++){
            if(vis[a[it]]) //这个数之前出现过
                add(vis[a[it]],-a[it]); //取消标记
            vis[a[it]]=it; //记录最大位置
            add(it,a[it]); //添加新标记
        }

改为如下代码即可

        for(;it<=q[i].R;it++){
            if(vis[a[it]]) //这个数之前出现过
                add(vis[a[it]],-1); //取消标记
            vis[a[it]]=it; //记录最大位置
            add(it,1); //添加新标记
        }

【代码】

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+10;
typedef long long ll;
struct p{
    int L,R,num;
}q[maxn];
int cmp(p a,p b){return a.R<b.R;}
ll sum[maxn]={0},ans[maxn]={0};
int n,vis[maxn]={0};
ll a[maxn];
void add(int p,ll y)
{
    while(p<=n){
        sum[p]+=y;
        p+=(p&-p);
    }
}
ll cx(int p)
{
    ll ans=0;
    while(p){
        ans+=sum[p];
        p-=(p&-p);
    }
    return ans;
}
int main()
{
    int m; scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].L,&q[i].R);
        q[i].num=i;
    }
    sort(q+1,q+1+m,cmp);
    int it=1;
    for(int i=1;i<=m;i++){
        for(;it<=q[i].R;it++){
            if(vis[a[it]]) //这个数之前出现过
                add(vis[a[it]],-a[it]); //取消标记
            vis[a[it]]=it; //记录最大位置
            add(it,a[it]); //添加新标记
        }
        it=q[i].R+1;
        ans[q[i].num]=cx(q[i].R)-cx(q[i].L-1); //查询
    }
    for(int i=1;i<=m;i++)
        printf("%lld\n",ans[i]);
    return 0;
}