POJ 2777 线段树基础题

题意：

给你一个长度为N的线段数，一开始每个树的颜色都是1，然后有2个操作。

第一个操作，将区间[a , b ]的颜色换成c。

第二个操作，输出区间[a , b ]不同颜色的总数。

直接线段树搞之。不过输入有个坑，a 可能大于b ，所以要判断一下。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2505
#define inf 1<<28
#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
#define PII pair<int,int>
using namespace std;

int L[1111111] , R[1111111] , data[1111111] , add[1111111] ;
int vis[1111] ;
int ans ;
int n , T ,O ;
void push_up(int x){
    if(data[LL(x)] != data[RR(x)])
        data[x] = 50 ;
    else
    data[x] = data[LL(x)] ;
}
void build(int l ,int r ,int u){
    L[u] = l ;
    R[u] = r ;
    data[u] = 1 ;
    add[u] = 0 ;
    if(l == r)return ;
    int mid = l + r >> 1 ;
    build(l ,mid ,LL(u)) ;
    build(mid + 1 ,r ,RR(u)) ;
}

void push_down(int x){
    if(add[x]){
        add[LL(x)] = add[x] ;
        add[RR(x)] = add[x] ;
        data[LL(x)] = add[x] ;
        data[RR(x)] = add[x] ;
        add[x] = 0 ;
    }
}

void updata(int l ,int r ,int u ,int color){
    push_down(u) ;
    if(l > R[u] || r < L[u])return ;
    if(l == L[u] && r == R[u]){
        data[u] = color ;
        add[u] = color ;
        return ;
    }
    int mid = (L[u] + R[u]) >> 1 ;
    if(r <= mid)updata(l , r , LL(u) ,color) ;
    else if(l > mid)updata(l ,r ,RR(u) ,color) ;
    else {
        updata(l ,mid ,LL(u), color) ;
        updata(mid + 1 ,r ,RR(u) ,color) ;
    }
    push_up(u) ;
}

void query(int l ,int r ,int u){
    push_down(u) ;
    if(l > R[u] || r < L[u])return ;
    if(l == L[u] && r == R[u]){
        if(data[u] != 50){
            if(!vis[data[u]]){
                ans ++ ;
                vis[data[u]] = 1 ;
            }
            return ;
        }
    }
    int mid = L[u] + R[u] >> 1 ;
    if(l > mid )query(l ,r , RR(u)) ;
    else if(r <= mid)query(l, r ,LL(u)) ;
    else {
        query(l , mid ,LL(u)) ;
        query(mid + 1 ,r ,RR(u)) ;
    }
    push_up(u) ;
}
inline void RD(int &ret) {
    char c;
    do {
        c = getchar();
    } while(c < '0' || c > '9') ;
    ret = c - '0';
    while((c=getchar()) >= '0' && c <= '9')
        ret = ret * 10 + ( c - '0' );
}
void swap(int& a ,int& b){
    int temp = a ;
    if(a > b){
        a = b ;
        b = temp ;
    }
}
void debug(){
    for (int i = 1 ;i <= 3 * n ;i ++){
        cout << "L:" << L[i] << "R:" << R[i] << "data:" << data[i] << "add:" << add[i] << endl;
    }
}
int main() {
    cin >> n >> T >> O ;
    build(1 , n , 1 ) ;
    int a , b , c ;
    char op[11] ;
    while(O -- ){
        scanf("%s",op) ;
        if(op[0] == 'C'){
            scanf("%d%d%d",&a,&b,&c) ;
            if( a > b)swap(a ,b ) ;
            updata(a , b, 1 , c) ;
        }
        else if(op[0] == 'P'){
            scanf("%d%d",&a,&b) ;
            if(a > b)swap(a , b) ;
            ans = 0 ;
            mem(vis,0) ;
            query(a , b , 1) ;
            printf("%d\n",ans) ;
        }
    }
    return 0 ;
}