POJ 1201 树状数组

链接：

http://poj.org/problem?id=1201

题意：

给你n个区间，每个区间为[a,b]，每个区间取c个数构成一个集合，求集合最小容量

题解：

把区间按b排序，从第一个区间开始取，从后往前取，这样尽可能和后面的区间重复

另外如果我们发现当前区间取得个数已经超过了c，那么只需要让之前区间换就行，而总数是不变的，所以不用更新答案

求当前区间已经取了多少个数用树状数组

代码：

#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define all(x) (x).begin(),(x).end()
#define pb push_back
#define mp make_pair
#define lson l,m,rt<<1  
#define rson m+1,r,rt<<1|1 
typedef long long ll;
typedef vector<int> VI;
typedef pair<int, int> PII;
const ll MOD = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int MAXN = 5e4 + 7;
// head

struct sec {
    int a, b, c;
    const bool operator<(const sec &t) const {
        if (b == t.b) return a < t.a;
        return b < t.b;
    }
}s[MAXN];

int n;
int bit[MAXN];
int used[MAXN];

void add(int i, int x) {
    while (i <= 5e4) bit[i] += x, i += i&-i;
}

int sum(int i) {
    int s = 0;
    while (i) s += bit[i], i -= i&-i;
    return s;
}

int main() {
    cin >> n;
    rep(i, 0, n) scanf("%d%d%d", &s[i].a, &s[i].b, &s[i].c);
    sort(s, s + n);
    int ans = 0;
    rep(i, 0, n) {
        int picked = sum(s[i].b) - sum(s[i].a - 1);
        if (picked < s[i].c) {
            int remain = s[i].c - picked;
            ans += remain;
            int tail = s[i].b;
            while (remain) {
                if (!used[tail]) {
                    used[tail] = 1;
                    add(tail, 1);
                    remain--;
                }
                tail--;
            }
        }
    }
    cout << ans << endl;
    return 0;
}