色板游戏 （珂朵莉树+优化）

色板游戏 - 洛谷

核心思路

用珂朵莉树处理区间赋值、计算颜色。

加一个记录答案，就可以过掉全部数据。

AC代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
struct ran{
    int l, r;
    mutable int v;
    bool operator <(const ran &x)const{//运算符重载,用于set的排序
        return l < x.l;//按区间的左端点从小到大来排
    }
};
#define It set<ran>::iterator
map<pair<int,pair<int,int>>,int> res;
set<ran>tr;//珂朵莉树的定义
It split(int pos){
    It it = tr.lower_bound({pos, -1, 0});//找到所需的pos的迭代器
    if(it != tr.end() && it->l == pos)return it;//看看这个迭代器的l是不是所需要的pos，是的话直接返回就行
    --it;//不是的话就说明肯定是在前一个里面
    int l = it->l;
    int r = it->r;
    ll v = it->v;
    tr.erase(it);//我们删掉这个区间
    tr.insert({l, pos - 1, v});//重新塞入两个区间[l, pos -1 ],[pos, r]
    return tr.insert({pos, r, v}).first;//返回以pos开头的区间的迭代器
}
bool pd[31];
void emerge(int l, int r, int v){
    It itr = split(r + 1);
    It itl = split(l);
  
    tr.erase(itl, itr);
    tr.insert(ran{l,r,v});
}
int t = 2;
int  ask(int l,int r){
	int ans = 0;
	It itr = split(r + 1);
    It itl = split(l);
	for(int i = 1;i <= t;i++){
		pd[i] = 0;
	}
	It it = itl;
    for(;it != itr;++it){
    	int col = it->v;
    //	cout<<col<<endl;
    	if(!pd[col]){
    		ans++;
    		pd[col] = 1;
		}
    }
    return ans;
}
int anss = 0;
int l,r;
int tot;
int main(){
	ios::sync_with_stdio(0);
	int L,O;
	cin>>L>>t>>O;
	tr.insert(ran{0,L+1,1});
	for(int i = 1;i <= O;i++){
		char op;
		cin>>op;
		if(op == 'C'){
			int l,r,v;
			cin>>l>>r>>v;
			if(r < l){
				swap(l,r);
			}
			++tot;
			emerge(l,r,v);
		}
		else{

			cin>>l>>r;
			if(r < l){
				swap(l,r);
			}
			if(res[{tot,{l,r}}]){
				cout<<res[{tot,{l,r}}]<<endl;
				continue;
			}
			int anser = ask(l,r);
			res[{tot,{l,r}}] = anser;
			cout<<anser<<endl;
		}
	}
	return 0;
}