HDU 3577 Fast Arrangement (线段树区间更新)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3577

题意不好理解，给你数字k表示这里车最多同时坐k个人，然后有q个询问，每个询问是每个人的上车和下车时间，每个人按次序上车，问哪些人能上车输出他们的序号。

这题用线段树的成段更新，把每个人的上下车时间看做一个线段，每次上车就把这个区间都加1，但是上车的前提是这个区间上的最大值不超过k。有个坑点就是一个人上下车的时间是左闭右开区间，可以想到要是一个人下车，另一个人上车，这个情况下这个点的大小还是不变的。还有注意格式...

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 1e6 + 5;
struct segtree {
    int l , r , sum , add;
}T[MAXN << 2];
int x[MAXN / 10] , y[MAXN / 10] , ans[MAXN / 10];

void init(int p , int l , int r) {
    int mid = (l + r) >> 1;
    T[p].l = l , T[p].r = r , T[p].add = 0;
    if(l == r) {
        T[p].sum = 0;
        return ;
    }
    init(p << 1 , l , mid);
    init((p << 1)|1 , mid + 1 , r);
    T[p].sum = max(T[p << 1].sum , T[(p << 1)|1].sum);
}

void updata(int p , int l , int r , int val) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(l == T[p].l && T[p].r == r) {
        T[p].sum += val;
        T[p].add += val;
        return ;
    }
    if(T[p].add) {
        T[p << 1].sum += T[p].add;
        T[p << 1].add += T[p].add;
        T[(p << 1)|1].sum += T[p].add;
        T[(p << 1)|1].add += T[p].add;
        T[p].add = 0;
    }
    if(r <= mid) {
        updata(p << 1 , l , r , val);
    }
    else if(l > mid) {
        updata((p << 1)|1 , l , r , val);
    }
    else {
        updata(p << 1 , l , mid ,val);
        updata((p << 1)|1 , mid + 1 , r , val);
    }
    T[p].sum = max(T[p << 1].sum , max(T[(p << 1)|1].sum , T[p].sum));
}

int query(int p , int l , int r) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(l == T[p].l && T[p].r == r) {
        return T[p].sum;
    }
    if(T[p].add) {
        T[p << 1].sum += T[p].add;
        T[p << 1].add += T[p].add;
        T[(p << 1)|1].sum += T[p].add;
        T[(p << 1)|1].add += T[p].add;
        T[p].add = 0;
    }
    if(r <= mid) {
        return query(p << 1 , l , r);
    }
    else if(l > mid) {
        return query((p << 1)|1 , l , r);
    }
    else {
        return max(query(p << 1 , l , mid) , query((p << 1)|1 , mid + 1 , r));
    }
}

int main() 
{
    int t , k , m;
    scanf("%d" , &t);
    for(int ca = 1 ; ca <= t ; ca++) {
        scanf("%d %d" , &k , &m);
        int len = -1;
        for(int i = 1 ; i <= m ; i++) {
            scanf("%d %d" , x + i , y + i);
            y[i]--;
            len = max(x[i] , len);
            len = max(y[i] , len);
        }
        init(1 , 1 , len);
        int cont = 0;
        for(int i = 1 ; i <= m ; i++) {
            int temp = query(1 , x[i] , y[i]);
            if(temp < k) {
                ans[++cont] = i;
                updata(1 , x[i] , y[i] , 1);
            }
        }
        printf("Case %d:
" , ca);
        for(int i = 1 ; i <= cont ; i++) {
            printf("%d " , ans[i]);
        }
        printf("

");
    }
}