【洛谷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的每次询问，输出一个答案（单独一行），表示当前区间地雷总数。

样例输入：

样例输出：

1
2

数据范围：

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

时间限制：

空间限制：

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\n",ask1(r)-ask2(l-1));
    }
    return 0;
}