线段树——校门外的树

版权声明：叶小小 https://blog.csdn.net/dy416524/article/details/86763212

题目描述

校门外有很多树，有苹果树，香蕉树，有会扔石头的，有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两个操作：
K=1，K=1，读入l、r表示在区间[l,r]中种上一种树，每次操作种的树的种类都不同
K=2，读入l,r表示询问l~r之间能见到多少种树
（l,r>0）

输入

第一行n,m表示道路总长为n，共有m个操作
接下来m行为m个操作

输出

对于每个k=2输出一个答案

样例输入

复制样例数据

5 4
1 1 3
2 2 5
1 2 4
2 3 5

样例输出

1
2

提示

20%的数据保证，n,m<=100
60%的数据保证，n <=1000,m<=50000
100%的数据保证，n,m<=50000

主要是可能有重复的树，不是取最大。

#include <iostream>
#include <cstdio>
#define N 200010

using namespace std;
int sum[N],lazy[N],ex[N];
int n,m;
void build(int l,int r,int rt){
    if(l==r){
        sum[rt]=0;
        return;
    }
    int mid=(l+r)/2;
    build(l,mid,rt*2);
    build(mid+1,r,rt*2+1);
    sum[rt]=sum[rt*2]+sum[rt*2+1];
}
void pushdown(int l,int r,int rt){
    if(lazy[rt]){
        ex[rt]+=lazy[rt];重复的种类
        sum[rt*2]+=lazy[rt];
        sum[rt*2+1]+=lazy[rt];
        lazy[rt*2]+=lazy[rt];
        lazy[rt*2+1]+=lazy[rt];
        lazy[rt]=0;
    }
}
void update(int l1,int r1,int l,int r,int rt){
    if(l1>r||r1<l) return;
    sum[rt]++;
    if(l1<=l&&r1>=r){
        lazy[rt]+=1;
        return ;
    }
    int mid=(l+r)/2;
    pushdown(mid-l+1,r-mid,rt);
    if(l1<=mid){
        update(l1,r1,l,mid,rt*2);
    }
    if(r1>mid){
        update(l1,r1,mid+1,r,rt*2+1);
    }
    if(r1>mid&&l1<=mid){
        ex[rt]++;//重复的种类
    }
}
int query(int l1,int r1,int l,int r,int rt){
    if(l1<=l&&r1>=r) {
        return sum[rt];
    }
    int mid=(r+l)/2;
    int ret=0;
    pushdown(mid-l+1,r-mid,rt);
    if(l1<=mid){
        ret+=query(l1,r1,l,mid,rt*2);
    }
    if(r1>mid){
        ret+=query(l1,r1,mid+1,r,rt*2+1);
    }
    if(r1>mid&&l1<=mid){
        ret-=ex[rt];//加了两遍，减一次
    }
    return ret;
}
int main()
{
    scanf("%d%d",&n,&m);
    build(1,n,1);
    while(m--){
        int k,a,b;
        scanf("%d%d%d",&k,&a,&b);
        if(k==1){
            update(a,b,1,n,1);
        }
        else{
            int ans=query(a,b,1,n,1);
            printf("%d\n",ans);
        }
    }
    return 0;
}