【洛谷P2184】贪婪大陆

题目链接

贪婪大陆

题目背景：

面对蚂蚁们的疯狂进攻，小FF的Tower defence宣告失败……人类被蚂蚁们逼到了Greed Island上的一个海湾。现在，小FF的后方是一望无际的大海， 前方是变异了的超级蚂蚁。 小FF还有大好前程，他可不想命丧于此， 于是他派遣手下最后一批改造SCV布置地雷以阻挡蚂蚁们的进攻。
小FF最后一道防线是一条长度为N的战壕， 小FF拥有无数多种地雷，而SCV每次可以在([ L , R ])区间埋放同一种不同于之前已经埋放的地雷。 由于情况已经十万火急，小FF在某些时候可能会询问你在([ L' , R']) 区间内有多少种不同的地雷， 他希望你能尽快的给予答复。

输入格式：

第一行为两个整数(n)和(m)； (n)表示防线长度， (m)表示SCV布雷次数及小FF询问的次数总和。
接下来有(m)行， 每行三个整数(Q)，(L) , (R)； 若(Q=1) 则表示SCV在([ L , R ])这段区间布上一种地雷， 若(Q=2)则表示小FF询问当前([ L , R ])区间总共有多少种地雷。

输出格式：

对于小FF的每次询问，输出一个答案（单独一行），表示当前区间地雷总数。

样例输入：

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

样例输出：

1
2

数据范围：

对于30%的数据： (0<=n, m<=1000);
对于100%的数据： (0<=n, m<=10^5).

时间限制：

1S

空间限制：

50M

提示：

remove!!!

题解

求(L)到(R)之间有多少种地雷，因为放地雷都是按照区间放的，而且每次放的地雷都不相同，
那么只要用1到(L)的地雷种数（也就是从1到(L)开始放置地雷的个数）再减去1到(R-1)已经放置结束了的地雷（也就是放置地雷时的(R)）种数就是答案了。
接下来就简单了，用两个树状数组维护两个前缀和就好了。
上代码：

#include<bits/stdc++.h>
using namespace std;
int n,m;
int q,l,r;
int a1[100009],a2[100009];
int lb(int x){
    return x&(-x);
}
void add1(int x){
    while(x<=n){
        a1[x]++;
        x+=lb(x);
    }
}
void add2(int x){
    while(x<=n){
        a2[x]++;
        x+=lb(x);
    }
}
int ask1(int x){
    int ans=0;
    while(x){
        ans+=a1[x];
        x-=lb(x);
    }
    return ans;
}
int ask2(int x){
    int ans=0;
    while(x){
        ans+=a2[x];
        x-=lb(x);
    }
    return ans;
}
int main(){
    scanf("%d%d",&n,&m);
    while(m--){
        scanf("%d%d%d",&q,&l,&r);
        if(q==1){
            add1(l);
            add2(r);
        }
        else if(q==2) printf("%d
",ask1(r)-ask2(l-1));
    }
    return 0;
}