hdu3577 Fast Arrangement（线段树，lazy区间更新）

Fast Arrangement

#include <bits/stdc++.h>
using namespace std;
struct Tree
{
	int l,r,v;//v,区间l,r间的最大购票数（最大上车人数）
	int lazy;
}b[4000000+1024];
int res[1000000];
void build(int l,int r,int i)
{
	b[i].l=l;
	b[i].r=r;
	b[i].v=0;
	b[i].lazy=0;
	if(l==r) return;
	int m=(l+r)/2;
	build(l,m,2*i);
	build(m+1,r,2*i+1);
}
void pushdownlazy(int i)
{
	b[2*i].lazy+=b[i].lazy;
	b[2*i+1].lazy+=b[i].lazy;
	b[2*i].v+=b[i].lazy;
	b[2*i+1].v+=b[i].lazy;
	b[i].lazy=0;//i节点清零
}
int query(int l,int r,int i)
{
	if(b[i].l==l&&b[i].r==r){
		return b[i].v;
	}
	if(b[i].lazy>0) pushdownlazy(i);
	int m=(b[i].l+b[i].r)/2;
	if(r<=m) return query(l,r,2*i);
	else if(l>m) return query(l,r,2*i+1);
	return max(query(l,m,2*i),query(m+1,r,2*i+1));
}
void update(int l,int r,int i)
{
	if(b[i].l==l&&b[i].r==r){
		b[i].lazy++;
		b[i].v++;
		return;
	}
	if(b[i].lazy) pushdownlazy(i);
	int m=(b[i].l+b[i].r)/2;
	if(r<=m) update(l,r,2*i);
	else if(l>m) update(l,r,2*i+1);
	else{
		update(l,m,2*i);
		update(m+1,r,2*i+1);
	}
	b[i].v=max(b[2*i].v,b[2*i+1].v);
}
int main()
{
	int t,cs=0;
	scanf("%d",&t);
	while(t--){
		int k,q,len=0;
		scanf("%d %d",&k,&q);
		build(1,1000000,1);
		for(int i=1;i<=q;++i){
			int l,r;
			scanf("%d %d",&l,&r);
			--r;
			if(query(l,r,1)<k){
				res[len++]=i;
				update(l,r,1);
			}
		}
		printf("Case %d:\n",++cs);
		for(int i=0;i<len;++i){
			printf("%d ",res[i]);
		}
		printf("\n\n");
	}
	return 0;
}