P2184 贪婪大陆

题目背景

面对蚂蚁们的疯狂进攻，小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的每次询问，输出一个答案（单独一行），表示当前区间地雷总数。

思路：

题意可以理解为询问一个区间中有多少段不同的区间，我们知道只要开头在i前，这个区间就在[1,i]中，只要结尾在i后，这个区间不再[1,i]中，所以若问[i,j]之间的地雷种数就是j之前的开头数目的和减去i之前不包括i的结尾数目和。两个线段树维护开头和结尾，很容易想出来。

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
#define MAX 100010
using namespace std;
ll n, m, w[MAX << 2], lz[MAX << 2], f, a, b, p[MAX << 2], lp[MAX << 2];
void up(ll rt) {
    w[rt] = w[rt << 1] + w[rt << 1 | 1];
}
void up1(ll rt) {
    p[rt] = p[rt << 1] + p[rt << 1 | 1];
}
void down1(ll rt, ll ln, ll rn) {
    if(lp[rt]) {
        lp[rt << 1] += lp[rt];
        lp[rt << 1 | 1] += lp[rt];
        p[rt << 1] += lp[rt] * ln;
        p[rt << 1 | 1] += lp[rt] * rn;
        lp[rt] = 0;
    }
}
void down(ll rt, ll ln, ll rn) {
    if(lz[rt]) {
        lz[rt << 1] += lz[rt];
        lz[rt << 1 | 1] += lz[rt];
        w[rt << 1] += lz[rt] * ln;
        w[rt << 1 | 1] += lz[rt] * rn;
        lz[rt] = 0;
    }
}
void update(ll L, ll R, ll C, ll l, ll r, ll rt) {
    if(L <= l && r <= R) {
        w[rt] += C * (r - l + 1);
        lz[rt] += C;
        return;
    }
    ll m = (r + l) >> 1;
    down(rt, m - l + 1, r - m);
    if(L <= m)
        update(L, R, C, l, m, rt << 1);
    if(R > m)
        update(L, R, C, m + 1, r, rt << 1 | 1);
    up(rt);
}
void update1(ll L, ll R, ll C, ll l, ll r, ll rt) {
    if(L <= l && r <= R) {
        p[rt] += C * (r - l + 1);
        lp[rt] += C;
        return;
    }
    ll m = (r + l) >> 1;
    down1(rt, m - l + 1, r - m);
    if(L <= m)
        update1(L, R, C, l, m, rt << 1);
    if(R > m)
        update1(L, R, C, m + 1, r, rt << 1 | 1);
    up1(rt);
}
ll query(ll L, ll R, ll l, ll r, ll rt) {
    if(L <= l && r <= R) {
        return w[rt];
    }
    ll m = (r + l) >> 1;
    down(rt, m - l + 1, r - m);
    ll ans = 0;
    if(L <= m)
        ans += query(L, R, l, m, rt << 1);
    if(R > m)
        ans += query(L, R, m + 1, r, rt << 1 | 1);
    return ans;
}
ll query1(ll L, ll R, ll l, ll r, ll rt) {
    if(L <= l && r <= R) {
        return p[rt];
    }
    ll m = (r + l) >> 1;
    down1(rt, m - l + 1, r - m);
    ll ans = 0;
    if(L <= m)
        ans += query1(L, R, l, m, rt << 1);
    if(R > m)
        ans += query1(L, R, m + 1, r, rt << 1 | 1);
    return ans;
}
int main() {
    cin >> n >> m;
    for(int i = 0; i < m; i++) {
        cin >> f;
        if(f == 1) {
            cin >> a >> b;
            update(a, a, 1, 1, n, 1);
            update1(b, b, 1, 1, n, 1);
        } else {
            cin >> a >> b;
            cout << query(1, b, 1, n, 1) - query1(1, a - 1, 1, n, 1) << endl;
        }
    }
    return 0;
}