Hdu 4614 线段树

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define N 50005
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define initial 0,n-1,1
#define pb push_back
#define mp make_pair
using namespace std;
int n,m;

struct segtree{
int  a0[N<<2],a1[N<<2],lazy[N<<2];
void up(int rt)
{
a0[rt]=a0[rt<<1]+a0[rt<<1|1];
a1[rt]=a1[rt<<1]+a1[rt<<1|1];
}
void build(int l,int r,int rt)
{   lazy[rt]=0;
if (l==r) {     a1[rt]=0;
a0[rt]=1;
return;
}

int mid=(l+r)>>1;
build(lson);
build(rson);
up(rt);
}

void fun(int rt,int op,int m)
{   if (op==1)
{      a0[rt]=m; a1[rt]=0; }
if (op==2)
{      a0[rt]=0; a1[rt]=m; }

lazy[rt]=op;
}
void down(int rt,int m)
{
if (lazy[rt])
{   fun(rt<<1,lazy[rt],m-(m>>1));
fun(rt<<1|1,lazy[rt],m>>1);
lazy[rt]=0;
}
}
void update(int L,int R,int op,int l,int r,int rt)
{   if (L<=l&&r<=R)
{   fun(rt,op,r-l+1);
return;
}
down(rt,r-l+1);
int mid=(l+r)>>1;
if (L<=mid) update(L,R,op,lson);
if (mid<R)  update(L,R,op,rson);
up(rt);
}
int find(int c,int l,int r,int rt)
{   if (l==r) return l;
down(rt,r-l+1);
int mid=(l+r)>>1;
int re;
if (a0[rt<<1]>=c) re=find(c,lson);else re=find(c-a0[rt<<1],rson);
up(rt);
return re;

}
int query(int L,int R,int op,int l,int r,int rt)
{   if (L<=l&&r<=R) {
if (op==0)return a0[rt];
if (op==1)return a1[rt];
}
down(rt,r-l+1);
int mid=(l+r)>>1;
int re=0;
if (L<=mid) re+=query(L,R,op,lson);
if (mid<R)  re+=query(L,R,op,rson);
up(rt);
return re;
}
}st;

void doit()
{   char ss[12];
int x,y,z;
scanf("%d%d",&n,&m);
st.build(initial);
while (m--)
{   scanf("%d%d%d",&x,&y,&z);
if (x==1)
{

int tmp,left,ans1,ans2;
if (y!=0) tmp=st.query(0,y-1,0,initial);else tmp=0;
left=st.query(y,n-1,0,initial);
if (left==0)  {printf("Can not put any one.\n"); continue;}
ans1=st.find(tmp+1,initial);
ans2=st.find(tmp+min(left,z),initial);
printf("%d %d\n",ans1,ans2);
st.update(ans1,ans2,2,initial);

}
if (x==2)
{   printf("%d\n",st.query(y,z,1,initial));
st.update(y,z,1,initial);
}
}

}

int main()
{   int cas;
scanf("%d",&cas);
while (cas--) {doit();printf("\n");}
return 0;
}