セグメントツリー、区間更新+オーラ関数

2474 ワード

区間更新は,新しい更新姿勢を習得した.
HDU 5634 
操作1、所与の区間、区間内の各単点、関数に従って変換
操作2、区間更新、区間全体のすべての値を修正
操作3,加算
#include 
#include 
#include 
#include  
#include 
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ll rt<<1
#define rr rt<<1|1
using namespace std;
typedef long long LL;
const int maxn=300001;
const int maxm=1e7+1;
long long ph[maxm];
// 
void Eor(){
	memset(ph,0,sizeof(ph));
	ph[1]=1;
	for(long long i=2;i>1;
	build(l,m,ll); build(m+1,r,rr);
	PushUP(rt);
}
void update1(int L,int R,int rt)
{
	if(tree[rt].lazy && L==tree[rt].l && R==tree[rt].r) {
		tree[rt].sum = ph[tree[rt].lazy] * tree[rt].len;
		tree[rt].lazy = ph[tree[rt].lazy];
		return ;
	}
	PushDown(rt);
	int m=(tree[rt].l+tree[rt].r)>>1;
	if(R <= m) update1(L,R,ll);
	else if(L > m) update1(L,R,rr);
	else { update1(L,m,ll); update1(m+1,R,rr);}
	PushUP(rt);
}
void update2(int L,int R,int neww,int rt)
{
	if(L==tree[rt].l && R==tree[rt].r) {
		tree[rt].sum = 1LL*neww*tree[rt].len;
		tree[rt].lazy = neww;
		return ;
	}
	PushDown(rt);
	int m=(tree[rt].l+tree[rt].r)>>1;
	if(R <= m) update2(L,R,neww,ll);
	else if(L > m) update2(L,R,neww,rr);
	else { update2(L,m,neww,ll); update2(m+1,R,neww,rr);}
	PushUP(rt);
}
long long query(int L,int R,int rt)
{
	if(L==tree[rt].l && R==tree[rt].r)  
		return tree[rt].sum;
	PushDown(rt);
	int m=(tree[rt].l+tree[rt].r)>>1;
	if(R <= m) return query(L,R,ll);
	else if(L > m) return query(L,R,rr);
	else return query(L,m,ll)+query(m+1,R,rr);
} 
int main()
{
	int T;
	int n,m;
	scanf("%d",&T);
	Eor();
	while(T--)
	{
		scanf("%d%d",&n,&m);
		build(1,n,1);
		int tp,l,x,r;
		while(m--){
			scanf("%d%d%d",&tp,&l,&r);
			if(tp==1) update1(l,r,1);
			else if(tp==2){ scanf("%d",&x); update2(l,r,x,1);}
			else if(tp==3) printf("%lld
",query(l,r,1)); } } return 0; }