关于线段树的结构体封装

struct segtree{
	int sum[N*350],ls[N*350],rs[N*350],cnt;
	void upd(int &p,int l,int r,int x,int v){
		if(!p)p=++cnt;
		sum[p]+=v;
		if(l==r)return ;
		int mid=(l+r)>>1;
		if(mid>=x)upd(ls[p],l,mid,x,v);
		else upd(rs[p],mid+1,r,x,v);
	}
	int qry(int p,int l,int r,int x,bool op){
		if(!p||!x||x>nn)return 0;
		if(l==r)return sum[p];
		int mid=(l+r)>>1;
		if(op==0){
			if(mid>=x)return qry(ls[p],l,mid,x,op);
			else return sum[ls[p]]+qry(rs[p],mid+1,r,x,op);
		}
		if(op==1){
			if(mid>=x)return qry(ls[p],l,mid,x,op)+sum[rs[p]];
			else return qry(rs[p],mid+1,r,x,op);
		}
	}
}s0;

struct node{
	int sum,ls,rs;
};
struct segtree{
	node tr[N*350];
	int cnt;
	void upd(int &p,int l,int r,int x,int v){
		if(!p)p=++cnt;
		tr[p].sum+=v;
		if(l==r)return ;
		int mid=(l+r)>>1;
		if(mid>=x)upd(tr[p].ls,l,mid,x,v);
		else upd(tr[p].rs,mid+1,r,x,v);
	}
	int qry(int p,int l,int r,int x,int y){
		if(x>y||!p)return 0;
		if(l>=x&&r<=y)return tr[p].sum;
		int mid=(l+r)>>1,ans=0;
		if(mid>=x)ans+=qry(tr[p].ls,l,mid,x,y);
		if(mid<y)ans+=qry(tr[p].rs,mid+1,r,x,y);
		return ans;
	}
}s0;