hdu 1698 Just a Hook(线段树 成段更新+总区间求和)

http://acm.hdu.edu.cn/showproblem.php?pid=1698

题意：给定一根棒的长度，它上面有价值1,2,3种，初始价值为1，每次输入的x,y,z表示把区间[x,y]的价值全部变为z；问最后这根棒的总价值是多少。

思路：与书中对某一线段[x,y]同时加上z类似，只不过这个是把线段[x,y]变成z。很经典的应用。

属于线段树成段更新问题。。成段更新与单点更新不同，成段更新没必要更新到单个点，当找到要更新的区间时直接更新这一整个区间返回即可，若上面没有返回，就要传递属性，即把该区间的价值属性传递给左右儿子，同时更新该节点的信息，并且把该区间的价值属性设为-1，标志它已把价值传递给左右子树了。最后更新完子树后要回溯更新这个节点的信息。

询问区间的和时，若正好询问到该区间并且价值和一定时直接返回价值和，若它的价值属性是-1要递归左右子树求和。否则将信息向下推送，直到找到相等的区间。

#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL long long
#define _LL __int64
#define eps 1e-12
#define PI acos(-1.0)
#define C 240
#define S 20
using namespace std;

const int maxn = 100010;

struct node
{
	int l,r;
	int val; //维护该区间的价值
	int sum;//维护该区间的价值和

}tree[maxn*4];

void build(int v, int l, int r)
{
	tree[v].l = l;
	tree[v].r = r;
	tree[v].val = 1;
	tree[v].sum = 1;
	if(l == r)
	{
		return;
	}
	int mid = (l+r) >> 1;
	build(v*2,l,mid);
	build(v*2+1,mid+1,r);
	tree[v].sum = tree[v*2].sum + tree[v*2+1].sum;
}

void update(int v, int l, int r, int val)
{
	if(tree[v].l == l && tree[v].r == r)
	{
		tree[v].val = val;
		tree[v].sum = val*(tree[v].r - tree[v].l + 1);
		return;
	}
	if(tree[v].val != -1)
	{
		int mid = (tree[v].l + tree[v].r) >> 1;
		tree[v].sum = tree[v].val*(tree[v].r - tree[v].l + 1); //向下推送同时更新该节点的价值和
		update(v*2,tree[v].l,mid,tree[v].val);
		update(v*2+1,mid+1,tree[v].r,tree[v].val);
		tree[v].val = -1; //价值置为1，因为向下推送了。
	}

	int mid = (tree[v].l + tree[v].r) >> 1;
	if(r <= mid)
		update(v*2,l,r,val);
	else if(l > mid)
		update(v*2+1,l,r,val);
	else
	{
		update(v*2,l,mid,val);
		update(v*2+1,mid+1,r,val);
	}
	tree[v].sum = tree[v*2].sum + tree[v*2+1].sum; //回溯更新该节点
}

int query(int v, int l, int r)
{
	if(tree[v].l == l && tree[v].r == r)
	{
		if(tree[v].val != -1)
			return tree[v].val * (tree[v].r - tree[v].l + 1);
		else 
		{ //若是-1，递归计算左右子树
			int mid = (tree[v].l + tree[v].r) >> 1;
			return query(v*2,l,mid) + query(v*2+1,mid+1,r);
		}
	}

	int mid = (tree[v].l + tree[v].r) >> 1;
    //上面没走掉，要将信息向下推送，同时将该节点价值置为-1
	if(tree[v].val != -1)
	{
		update(v*2,tree[v].l,mid,tree[v].val);
		update(v*2+1,mid+1,tree[v].r,tree[v].val);
		tree[v].val = -1;
	}

	if(r <= mid)
		return query(v*2,l,r);
	else if(l > mid)
		return query(v*2+1,l,r);
	return query(v*2,l,mid) + query(v*2+1,mid+1,r);
}

int main()
{
	int test;
	int n,m;
	int l,r,val;

	scanf("%d",&test);
	for(int item = 1; item <= test; item++)
	{
		scanf("%d",&n);
		build(1,1,n);
		scanf("%d",&m);
		while(m--)
		{
			scanf("%d %d %d",&l,&r,&val);
			update(1,l,r,val);
		}
		printf("Case %d: The total value of the hook is %d.\n",item,query(1,1,n));
	}
	return 0;
}

#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL long long
#define _LL __int64
#define eps 1e-12
#define PI acos(-1.0)
#define C 240
#define S 20
using namespace std;

const int maxn = 100010;

struct node
{
	int l,r;
	int val; //维护该区间的价值

}tree[maxn*4];

void build(int v, int l, int r)
{
	tree[v].l = l;
	tree[v].r = r;
	tree[v].val = 1;
	if(l == r)
	{
		return;
	}
	int mid = (l+r) >> 1;
	build(v*2,l,mid);
	build(v*2+1,mid+1,r);
}

void update(int v, int l, int r, int val)
{
	if(tree[v].l == l && tree[v].r == r)
	{
		tree[v].val = val;
		return;
	}
	if(tree[v].val != -1)
	{
		tree[v*2].val = tree[v].val;
		tree[v*2+1].val = tree[v].val;
		tree[v].val = -1; //价值置为1，因为向下推送了。
	}

	int mid = (tree[v].l + tree[v].r) >> 1;
	if(r <= mid)
		update(v*2,l,r,val);
	else if(l > mid)
		update(v*2+1,l,r,val);
	else
	{
		update(v*2,l,mid,val);
		update(v*2+1,mid+1,r,val);
	}
}

int query(int v, int l, int r)
{
	if(tree[v].val != -1)
	{
		return tree[v].val * (r - l + 1);
	}

	int mid = (tree[v].l + tree[v].r) >> 1;

	if(r <= mid)
		return query(v*2,l,r);
	else if(l > mid)
		return query(v*2+1,l,r);
	else
		return query(v*2,l,mid) + query(v*2+1,mid+1,r);
}

int main()
{
	int test;
	int n,m;
	int l,r,val;

	scanf("%d",&test);
	for(int item = 1; item <= test; item++)
	{
		scanf("%d",&n);
		build(1,1,n);
		scanf("%d",&m);
		while(m--)
		{
			scanf("%d %d %d",&l,&r,&val);
			update(1,l,r,val);
		}
		printf("Case %d: The total value of the hook is %d.\n",item,query(1,1,n));
	}
	return 0;
}