一本通1537校门外的树

 　

1537：【例 3】校门外的树

时间限制: 1000 ms         内存限制: 524288 KB

【题目描述】

原题来自：Vijos P1448

校门外有很多树，学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两种操作：

K=1K=1，读入 l,rl,r 表示在 ll 到 rr 之间种上一种树，每次操作种的树的种类都不同；

K=2K=2，读入 l,rl,r 表示询问 ll 到 rr 之间有多少种树。

注意：每个位置都可以重复种树。

【输入】

第一行 n,mn,m 表示道路总长为 nn，共有 mm 个操作；

接下来 mm 行为 mm 个操作。

【输出】

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

【输入样例】

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

【输出样例】

1
2

【提示】

数据范围与提示：

对于 20% 的数据，1≤n,m≤1001≤n,m≤100；

对于 60% 的数据，1≤n≤10^3,1≤m≤5×10^4

对于 100% 的数据，1≤n,m≤5×1
1≤n,m≤5×104 ，保证 l,r>0

 

要使两个区间有交集，如[a,b]与[s,t] 要满足 a<=t且b>=s

 

于是根据本题，维护两个BIT即可

#include <bits/stdc++.h>
using namespace std;
const int N=50005;
int n,m;
struct BIT
{
    int S[N];
    inline int lowbit(int x)
    {
        return ((x)&(-x));
    }
    inline void Ins(int Pos,int Tag)
    {
        while(Pos>0)
        {
            S[Pos]+=Tag;
            Pos-=lowbit(Pos);
        }
        return;
    }
    inline int Que(int Pos)
    {
        int Sum=0;
        while(Pos<=n)
        {
            Sum+=S[Pos];
            Pos+=lowbit(Pos);
        }
        return Sum;
    }
}T1,T2;
int main()
{
//    freopen("tree2.in","r",stdin);
//    freopen("my.out","w",stdout);
    int i;
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++)
    {
        int opt,a,b;
        scanf("%d%d%d",&opt,&a,&b);
        if(opt==1)
        {
            T1.Ins(b,1);
            T2.Ins(a,1);
        }
        else
        {
            printf("%d
",T1.Que(a)-T2.Que(b+1));
        }
    }
    return 0;
}
/*
input
5 4
1 1 3
2 2 5
1 2 4
2 3 5
output
1
2
*/