【题】【线段树】NKOJ 1317 校门外有很多树

NKOJ 1317 校门外有很多树
时间限制 : 10000 MS 空间限制 : 65536 KB
问题描述
校门外有很多树，有苹果树，香蕉树，有会扔石头的，有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两个操作：
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

来源 vijos

思路：
每种一次树，树的种类加一，对于每次询问，总种类数减去不在询问区间内的种类即为所求。
也就是求已出现区间中有多少个完全不于该区间重合的区间。
ve表示该线段包含了多少个区间的右端点
vs表示该线段包含了多少个区间的左端点
每次询问时统计该区间左边有多少其他区间的右端点，右边有多少个其他区间的左端点
代码：

#include<iostream>
#include<cstdio>
using namespace std;
const int need=100004;//数据有鬼，数组要开大 

struct fy{int a,b,vs,ve;}t[need*4];

int x,y;

void build(int s,int x,int y)
{
    t[s].a=x,t[s].b=y;
    if(x==y) return;
    build(s<<1,x,(x+y)/2),build((s<<1)|1,(x+y)/2+1,y);
}

void add_s(int s)
{
    t[s].vs++;
    if(t[s<<1].a<=x&&x<=t[s<<1].b) add_s(s<<1);
    else if(t[(s<<1)|1].a<=x&&x<=t[(s<<1)|1].b) add_s((s<<1)|1);
}
void add_e(int s)
{
    if(t[s].a<=y&&y<=t[s].b) t[s].ve++;
    if(t[s<<1].a<=y&&y<=t[s<<1].b) add_e(s<<1);
    else if(t[(s<<1)|1].a<=y&&y<=t[(s<<1)|1].b) add_e((s<<1)|1);
}
//add_s和add_e不能写在一起 
int cnt_s(int s) //y~n
{
    if(t[s].b<y) return 0;
    if(y<=t[s].a) return t[s].vs;
    return cnt_s(s<<1)+cnt_s((s<<1)|1);
}
int cnt_e(int s) //1~x
{
    if(t[s].a>x) return 0;
    if(x>=t[s].b) return t[s].ve;
    return cnt_e(s<<1)+cnt_e((s<<1)|1);
}

int main()
{
    int n,m;scanf("%d%d",&n,&m);
    build(1,1,n);
    for(int i=1,a,tot=0;i<=m;i++)
    {
        scanf("%d%d%d",&a,&x,&y);
        if(a==1) 
        {
            tot++;
            add_s(1),add_e(1);
        }
        else 
        {
            x--,y++;//要计算区间外的 
            printf("%d\n",tot-cnt_s(1)-cnt_e(1));
        }
    }
}