Codeforces Round #590 D. Distinct Characters Queries

CF上给的标签是数据结构。但给的题解里的方法是很巧的暴力，用vector<set>维护每个字母出现的下标，每次修改加下标，擦下标。每次询问对每个字母分别lower_bound查找区间内是否存在这样位置，实在太妙了！

先看题解的方法。

 #include <bits/stdc++.h>
 #define debug(x) cout << #x << ": " << x << endl
 using namespace std;
 typedef long long ll;
 const int MAXN=2e5+;
 const int INF=0x3f3f3f3f;

 int main()
 {
     ios::sync_with_stdio(false);
     cin.tie();
     string s;
     cin>>s;
     vector<set<int> > alp();
     for(int i=;i<s.size();++i)
     {
         alp[s[i] - 'a'].insert(i);
     }
     int q;
     cin>>q;
     while(q--)
     {
         int op;
         cin>>op;
         if(op==)
         {
             int pos;
             char c;
             cin>>pos>>c;
             --pos;
             alp[s[pos]-'a'].erase(pos);
             s[pos]=c;
             alp[s[pos]-'a'].insert(pos);
         }
         else
         {
             int l,r;
             cin>>l>>r;
             --l,--r;
             int cnt=;
             for(int i=;i<;++i)
             {
                 auto k=alp[i].lower_bound(l);
                 if(k!=alp[i].end() && *k<=r) cnt++;
             }
             cout<<cnt<<endl;
         }
     }
     return ;
 }

然而我看别人线段树的代码没看懂。

但巧了我又去补了道 POJ - 2777

立马发现这两题有异曲同工之处。

POJ的题意是给线段染色，起始一种颜色，区间染色，最多30种颜色，询问区间中有几种颜色。应该是用了bitmask，用不同的位代表不同的颜色，每位上1代表存在这样的颜色，最后对所询问区间内按位或，再统计几位为1就为答案。

还要求你支持区间修改，这CF上这题只有要求单点更新的，很显然我们也可以用不同的位上的1代表不同的字母，最后将区间内的值按位或即可，最后统计有几位为1，就是这个区间几个不同的字母。

下见代码该题线段树解法。

 #include <iostream>
 #define debug(x) cout << #x << ": " << x << endl
 #define lson (rt<<1)
 #define rson (rt<<1|1)
 #define Lson l,m,lson
 #define Rson m+1,r,rson
 using namespace std;
 typedef long long ll;
 const int MAXN=1e5+;
 const int INF=0x3f3f3f3f;
 int seg[MAXN<<],lazy[MAXN<<];
 string s;

 inline void pushup(int rt){seg[rt]=(seg[lson]|seg[rson]);}

 void pushdown(int rt)
 {
     if(lazy[rt])
     {
         seg[lson]=seg[rson]=lazy[rt];
         lazy[lson]=lazy[rson]=lazy[rt];
         lazy[rt]=;
     }
 }

 void build(int rt,int l,int r)
 {
     if(l==r) {seg[rt]=<<(s[l-]-'a');return;}
     int mid=l+r>>;
     build(lson,l,mid);
     build(rson,mid+,r);
     pushup(rt);
 }

 void update(int rt,int l,int r,int L,int R,int item)
 {
     if(l>=L && r<=R) {seg[rt]=<<item;lazy[rt]=<<item;return;}
     int mid=l+r>>;
     pushdown(rt);
     if(L<=mid) update(lson,l,mid,L,R,item);
     if(R>mid) update(rson,mid+,r,L,R,item);
     pushup(rt);
 }

 int query(int rt,int l,int r,int L,int R)
 {
     if(l>=L && r<=R){return seg[rt];}
     int mid=l+r>>;
     pushdown(rt);
     int ans1=,ans2=,ans=;
     if(L<=mid) ans1=query(lson,l,mid,L,R);
     if(R>mid) ans2=query(rson,mid+,r,L,R);
     ans=ans1|ans2;
     return ans;

 }
 int main()
 {
     ios::sync_with_stdio(false);
     cin.tie();
     int q;
     cin>>s>>q;
     int n=s.size();
     build(,,n);
     while(q--)
     {
         int op;
         cin>>op;
         if(op==)
         {
             int p;
             string z;
             cin>>p>>z;
             int x=z[]-'a';
             update(,,n,p,p,x);
         }
         else if(op==)
         {
             int L,R;
             cin>>L>>R;
             int cnt=,ans=query(,,n,L,R);
             while(ans)
             {
                 if(ans&) cnt++;
                 ans>>=;
             }
             cout<<cnt<<endl;
         }
     }
     return ;
 }