HDU 3577 线段树区间更新

点击打开链接

题意：一辆火车，同一时间最多可以坐K个人，然后n个人依次买票，上车时间为a，下车时间为b，问这个人能不能上去，最后将所有的可以上车的人输出

思路：将所有的时间看成区间，然后对于a到b，询问这段中被覆盖的最大次数，若不超过K，说明这个人可以上这段，然后更新就好了

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=1000010;
int num[maxn*4],num1[maxn*4],n,m;
void pushdown(int node){
    if(num1[node]){
        num1[node<<1]+=num1[node];
        num1[node<<1|1]+=num1[node];
        num[node<<1]+=num1[node];
        num[node<<1|1]+=num1[node];
        num1[node]=0;
    }
}
void update(int d,int a,int b,int le,int ri,int node){
    if(a<=le&&ri<=b){
        num1[node]+=d;
        num[node]+=d;
        return ;
    }
    pushdown(node);
    int t=(le+ri)>>1;
    if(a<=t) update(d,a,b,le,t,node<<1);
    if(b>t) update(d,a,b,t+1,ri,node<<1|1);
    num[node]=max(num[node<<1],num[node<<1|1]);
}
int query(int a,int b,int le,int ri,int node){
    if(a<=le&&ri<=b) return num[node];
    pushdown(node);
    int t=(le+ri)>>1,ans1=0;
    if(a<=t) ans1=max(ans1,query(a,b,le,t,node<<1));
    if(b>t) ans1=max(ans1,query(a,b,t+1,ri,node<<1|1));
    return ans1;
}
int ans[maxn];
int main(){
    int T,cas=1,a,b;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        memset(num,0,sizeof(num));
        memset(num1,0,sizeof(num1));
        int k=0;
        for(int i=1;i<=m;i++){
            scanf("%d%d",&a,&b);
            int len=query(a,b-1,1,1000000,1);
            if(len<n){
                ans[k++]=i;
                update(1,a,b-1,1,1000000,1);
            }
        }
        printf("Case %d:\n",cas++);
        for(int i=0;i<k;i++) printf("%d ",ans[i]);
        printf("\n\n");
    }
    return 0;
}