luogu 2184 贪婪大陆 (树状数组)

题目背景

面对蚂蚁们的疯狂进攻，小FF的Tower defence宣告失败……人类被蚂蚁们逼到了Greed Island上的一个海湾。现在，小FF的后方是一望无际的大海，前方是变异了的超级蚂蚁。小FF还有大好前程，他可不想命丧于此，于是他派遣手下最后一批改造SCV布置地雷以阻挡蚂蚁们的进攻。

题目描述

小FF最后一道防线是一条长度为N的战壕，小FF拥有无数多种地雷，而SCV每次可以在[ L , R ]区间埋放同一种不同于之前已经埋放的地雷。由于情况已经十万火急，小FF在某些时候可能会询问你在[ L' , R'] 区间内有多少种不同的地雷，他希望你能尽快的给予答复。

对于30%的数据： 0<=n, m<=1000;

对于100%的数据： 0<=n, m<=10^5.

输入输出格式

输入格式：

第一行为两个整数n和m； n表示防线长度， m表示SCV布雷次数及小FF询问的次数总和。

接下来有m行，每行三个整数Q，L , R；若Q=1 则表示SCV在[ L , R ]这段区间布上一种地雷，若Q=2则表示小FF询问当前[ L , R ]区间总共有多少种地雷。

输出格式：

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

输入输出样例

输入样例#1： 复制

输出样例#1： 复制

1
2

题目链接：https://www.luogu.org/problemnew/show/P2184

题目分析：对左端点和右端点分别单点更新，容易看出，答案就是查询区间右端点左边的左端点个数 - 查询区间左端点左边的右端点个数

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int const MAX = 1e5 + 5;
int n, m;
int lsum[MAX], rsum[MAX];

int lowbit(int x) {
    return x & (-x);
}

void updateL(int pos, int val) {
    for (int i = pos; i <= n; i += lowbit(i)) {
        lsum[i] += val;
    }
}

void updateR(int pos, int val) {
    for (int i = pos; i <= n; i += lowbit(i)) {
        rsum[i] += val;
    }
}

int getLsum(int pos) {
    int ans = 0;
    for (int i = pos; i > 0; i -= lowbit(i)) {
        ans += lsum[i];
    }
    return ans;
}

int getRsum(int pos) {
    int ans = 0;
    for (int i = pos; i > 0; i -= lowbit(i)) {
        ans += rsum[i];
    }
    return ans;
}

int main() {
    scanf("%d %d", &n, &m);
    int q, l, r;
    while (m--) {
        scanf("%d %d %d", &q, &l, &r);
        if (q == 1) {
            updateL(l, 1);
            updateR(r, 1);
        } else {
            printf("%d\n", getLsum(r) - getRsum(l - 1));
        }
    }
}