Dynamic len(set(a[L:R])) UVA - 12345

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3767

动态查询区间内有多少不同颜色

用set维护每种颜色出现的位置 set每次插入或擦除后仍满足单调性

对于[l,r]区间内有多少不同颜色 就看有多少位置i 其上颜色clr[i]之前出现的位置小于l 这样就讲问题转换为区间加和问题了 主要是修改操作时要照顾前后造成的影响比较麻烦

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+10;
const int maxq=5e4+10;
const int maxm=1e6+10;

struct node
{
    int l,r,val;
};

set <int> st[maxm];
node tree[200*maxn];
int pos[maxm],ary[maxn],root[maxn],groot[maxn];
int n,q,num;

int lowbit(int x)
{
    return x&(-x);
}

int build(int l,int r)
{
    int cur,m;
    cur=num++;
    tree[cur].l=0,tree[cur].r=0,tree[cur].val=0;
    if(l==r) return cur;
    m=(l+r)/2;
    tree[cur].l=build(l,m);
    tree[cur].r=build(m+1,r);
    return cur;
}

int updateII(int rot,int tar,int val,int l,int r)
{
    int cur,m;
    cur=num++;
    tree[cur]=tree[rot];
    tree[cur].val+=val;
    if(l==r) return cur;
    m=(l+r)/2;
    if(tar<=m) tree[cur].l=updateII(tree[rot].l,tar,val,l,m);
    else tree[cur].r=updateII(tree[rot].r,tar,val,m+1,r);
    return cur;
}

void updateI(int pos,int tar,int val)
{
    int i;
    for(i=pos;i<=n;i+=lowbit(i)) groot[i]=updateII(groot[i],tar,val,0,n);
}

int queryII(int rot,int pl,int pr,int l,int r)
{
    int res,m;
    if(pl<=l&&r<=pr) return tree[rot].val;
    res=0,m=(l+r)/2;
    if(pl<=m) res+=queryII(tree[rot].l,pl,pr,l,m);
    if(pr>m) res+=queryII(tree[rot].r,pl,pr,m+1,r);
    return res;
}

int queryI(int l,int r,int pl,int pr)
{
    int res,i;
    res=queryII(root[r],pl,pr,0,n)-queryII(root[l],pl,pr,0,n);
    for(i=l;i>=1;i-=lowbit(i)) res-=queryII(groot[i],pl,pr,0,n);
    for(i=r;i>=1;i-=lowbit(i)) res+=queryII(groot[i],pl,pr,0,n);
    return res;
}

int main()
{
    set <int> ::iterator it;
    int i,p,v,l,r;
    char op[10];
    scanf("%d%d",&n,&q);
    for(i=1;i<=n;i++) scanf("%d",&ary[i]);
    num=0;
    root[0]=build(0,n);
    for(i=1;i<=1e6;i++) st[i].insert(0),st[i].insert(n+1);
    for(i=1;i<=n;i++){
        st[ary[i]].insert(i);
        root[i]=updateII(root[i-1],pos[ary[i]],1,0,n);
        pos[ary[i]]=i;
    }
    for(i=0;i<=n;i++) groot[i]=root[0];
    while(q--){
        scanf("%s",op);
        if(op[0]=='M'){
            scanf("%d%d",&p,&v);
            p++;

            it=st[ary[p]].lower_bound(p);
            l=*(--it);
            it++;
            r=*(++it);
            //printf("*%d %d %d*\n",l,r,p);
            updateI(p,l,-1),updateI(r,p,-1),updateI(r,l,1);
            st[ary[p]].erase(p),st[v].insert(p);
            ary[p]=v;
            it=st[v].lower_bound(p);
            l=*(--it);
            it++;
            r=*(++it);
            //printf("*%d %d %d*\n",l,r,p);
            updateI(p,l,1),updateI(r,p,1),updateI(r,l,-1);

        }
        else{
            scanf("%d%d",&l,&r);
            printf("%d\n",queryI(l,r,0,l));
        }
    }
    return 0;
}