线段树入门题目

hdu1698

#include <bits/stdc++.h>
using namespace std;

#define lson l,m,k<<1
#define rson m+1,r,k<<1|1
#define N 100001

int rt[N<<2];
int color; //代表当前更新是什么颜色 

void down( int k ){
    rt[k<<1] = rt[k<<1|1] = rt[k];
    rt[k] = 0;
}

void build( int l, int r, int k ){
    rt[k] = color;
    if( l==r ) return;
    int m = (l+r)>>1;
    build(lson);
    build(rson);
}

void update( int L, int R, int l, int r, int k ){
    if( L<=l&&R>=r ){
        rt[k] = color;
        return;
    }
    /*
	如果这次更新的总区间包含这段，
	那么就去更新子区间并重置该区间 
	*/ 
    if( rt[k] ){ 
        down(k);
    }
    int m = (l+r)>>1;
    if( L <= m ){
        update( L, R, lson );
    }
    if( R > m ){
        update( L, R, rson );
    }
}

int query( int l, int r, int k ){
	/*
	统计时 如果这段区间为同一个属性（题目中的材料）
	那么就可以将这段区间的总值计算出来，无需向下统计 
	*/ 
    if( rt[k] ){
        return rt[k] * ( r-l+1 );
    }
    int m = (r+l)>>1;
    return query(lson)+query(rson);
}

int main(){
    int T, n, t = 1;
    int L, R, q;
    scanf("%d",&T);
    while( T-- ){
        scanf("%d%d",&n,&q);
        color= 1; //初始时材料为铜（值为1，可以将材料看做一种颜色便于理解） 
        build(1,n,1);
        while( q-- ){
            scanf("%d%d%d",&L,&R,&color);
            update(L,R,1,n,1);
        }
        printf("Case %d: The total value of the hook is %d.\n",t++,query(1,n,1));
    }
    return 0;
}

POJ2777

#include <bits/stdc++.h>
using namespace std;
#define N 100005
int ans;
struct node{
	int left, right;
	int color;
	bool lazy;
	int mid(){
		return (left+right)>>1;
	}
}ltree[N<<2];

void build( int rt, int l, int r ){
	ltree[rt].left = l;
	ltree[rt].right = r;
	ltree[rt].color = 1;
	ltree[rt].lazy = false;
	if( l==r ) return;
	else{
		int mid = ltree[rt].mid();
		build( rt<<1,l,mid );
		build( rt<<1|1,mid+1,r );
	}
}

void update( int rt, int l, int r, int c ){
	/*
	1如果是这段区间，那么就加上lazy标记 
	2如果不是这段区间，lazy标记存在时将它向下传递
	并更新当前区间的左右区间 ，同时去除自身的lazy标记
	3如果根据需要更新的区间所在位置，向下更新
	4统计左右区间存在的颜色 
	*/
	if( ltree[rt].left==l&<ree[rt].right==r ){
		ltree[rt].color = c;
		if( l!=r ) ltree[rt].lazy = true;
		return;
	}else{
		if( ltree[rt].lazy ){
			ltree[rt<<1].lazy = ltree[rt<<1|1].lazy = true;
			ltree[rt<<1].color = ltree[rt<<1|1].color = ltree[rt].color; 
			ltree[rt].lazy = false; 
		}
		int mid = ltree[rt].mid();
		if( r<=mid ){
			update( rt<<1, l, r, c );
		}else if( l>mid ){
			update( rt<<1|1, l, r, c );
		}else{
			update( rt<<1, l, mid, c );
			update( rt<<1|1, mid+1, r, c ); 
		}
	}
	ltree[rt].color = ltree[rt<<1].color | ltree[rt<<1|1].color;
}

int query( int rt, int l, int r ){
	/*
	如果正好是当前区间或者是更大被lazy标记的范围，那么直接返回颜色数
	否则去向下统计 
	*/
	if( (ltree[rt].left==l&<ree[rt].right==r)||ltree[rt].lazy==1 ){
		return ltree[rt].color;
	}else{
		int mid = ltree[rt].mid();
		if( r <= mid ){
			return query( rt<<1, l, r ); 
		}else if( l > mid ){
			return query( rt<<1|1, l, r );
		}else{
			return query( rt<<1, l, mid ) | query( rt<<1|1, mid+1, r );
		}
	}
} 
int encode(int x){
	return 1<<(x-1);
}
int decode(int x){
	int t = 0;
	while( x > 0 ){
		if( x&1 ) t++;
		x >>= 1;
	} 
	return t;
}
int main(){
	int i, L, m, t;
	char c;
	int n1, n2, n3;
	while( ~scanf("%d%d%d",&L,&t,&m) ){
		build( 1,1,L );
		for( i = 0; i < m; i++ ){
			getchar();
			scanf("%c ",&c);
			if( c=='C' ){
				scanf("%d%d%d",&n1,&n2,&n3);
				n3 = encode(n3);
				if( n1 < n2 ){
					update( 1, n1, n2, n3 );
				}else{
					update( 1, n2, n1, n3 );
				}
			} else{
				scanf("%d%d",&n1,&n2);
				if( n1 < n2 ){
					ans = decode( query( 1,n1,n2 ) );
				}else{
					ans = decode( query( 1,n2,n1 ) );
				}
				printf("%d\n",ans); 
			}
		}
	}
	return 0;
}