线段树（成段更新）hdu 1698 Just a Hook

根据大神博客NotOnlySuccess的讲解，这个题目应该是属于线段树应用的第二阶段，也就是“成段更新”。大神博客已经有代码了，可能由于本题目不是很难所以没有过多的注释，如果是新手还是不容易看懂。所以这里就在啰嗦讲解一点吧。

题目：hdu 1698 Just a Hook

题目大意：dota中的屠夫（英雄）有一个钩子，钩子是连续的，现在假设钩子编号是从1到n，每个钩子有三种状态，分别对应的值是1、2、3（初始值都是1）。现在有m组操作：x，y，change；表示下标从x到y的钩子的值变为change。当然change的值只能是1、2、3。最后问这n个钩子的值的和sum是多少。

解题思路：首先建立线段树，每个断点至少记录一下当前这个区间的值，假设用key来记录，那么如果是1或者2或者3，就表示着段区间的所以值都是key。如果这段区间的值不一样那么就把key赋为-1。所以初始化应该都是1。对于怎么求sum，可以每次都updata_up向上去更新，这样就符合大多数线段树的思想了（对应代码1），但是对于这个题目，没有必要每次都向上去更新，可以等到所以操作完成后去遍历一下整个树就可以了（对应代码2）。

具体怎么实现可以参考一下代码：

hint：根据前辈们的教诲，程序最好写的整齐一些，这个不仅是指代码字母的整齐，还有布局和逻辑思想的整齐。

代码1：

//标准版
#include <iostream>
#include <cstdio>

using namespace std;

typedef struct RMQ{
	struct RMQ *lc, *rc;
	int left, right, key, sum;
	RMQ(){
	key=1;
	}
}*rmq;

rmq great(int x, int y){
	rmq q = new RMQ;
	q->left = x;
	q->right = y;
	q->sum = y - x + 1;
	if(x == y){
		q->lc = q->rc = NULL;
		return q;
	}
	q->lc = great(x,(x+y)>>1);
	q->rc = great(((x+y)>>1)+1,y);
	return q;
}

void update_down(rmq q){
	if(q ->key != -1){
		q ->lc ->key = q ->rc ->key =q ->key;
		q ->lc ->sum = q ->lc ->key * (q ->lc ->right - q ->lc ->left + 1);
		q ->rc ->sum = q ->rc ->key * (q ->rc ->right - q ->rc ->left + 1);
		q ->key = -1;
	}
}

void update_up(rmq q){
	q ->sum = q ->lc ->sum + q ->rc ->sum;
}

void update(rmq q, int x, int y, int change){
	if(x <= q->left && y>= q->right){
		q->key = change;
		q->sum = change  * (q ->right - q ->left + 1);
		return;
	}
	int mid = (q ->left + q->right)>>1;
	update_down(q);
	if(x <=mid ) update(q->lc, x, y, change);
	if(y > mid) update (q->rc, x, y, change);
	update_up(q);
}

int main(){
	int T, cas=1;
	cin >> T;
	while( T -- )
	{
		int n, m, x, y, change;
		cin >> n >>m;
		rmq head = new RMQ;
		head = great(1, n);
		head ->sum = n;
		for(int i = 0; i < m; i ++){
			cin >> x >> y >> change;
			update(head,x, y, change);
		}
		printf("Case %d: The total value of the hook is %d.\n",cas++,head ->sum);
	}
}

代码2：

//简洁版
#include <iostream>
#include <cstdio>

using namespace std;

typedef struct RMQ{
    struct RMQ *lc, *rc;
    int left, right, key, sum;
    RMQ(){
        key = 1;
    }
}*rmq;

rmq great(int x,int y){
    rmq q = new RMQ;
    q ->left = x;
    q ->right = y;
    if(x == y){
        q ->lc = q ->rc = NULL;
        return q;
    }
    int mid = (x + y) >> 1;
    q ->lc = great(x, mid);
    q ->rc = great(mid + 1, y);
    return q;
}

void update_down(rmq q){
    if(q ->key!=-1){
        q ->lc ->key = q ->rc ->key = q ->key;
        q ->key = -1;
    }
}
void update(rmq q, int x, int y, int key){
    if(x <= q ->left && y >= q ->right){
        q ->key = key;
        return ;
    }
    update_down(q);
    int mid = (q ->left + q ->right) >> 1;
    if(x <= mid) update(q ->lc, x, y, key);
    if(y > mid) update(q ->rc, x, y, key);
}

int fun(rmq q){
    if(q != NULL){
        if(q ->key != -1)
            return q ->key * (q ->right - q ->left + 1);
        else return fun(q ->lc) + fun(q ->rc);
    }
    else return 0;
}


int main(){
    int T,cas=1;
    cin >> T;
    while(T --){
        int n, m, x, y, change;
        cin >> n >> m;
        rmq head = new RMQ;
        head = great(1,n);
        for(int i = 0; i < m; i ++){
            cin >> x >> y >> change;
            update(head, x, y, change);
        }
        printf("Case %d: The total value of the hook is %d.\n",cas++,fun(head));
    }
}

此文章有nyist_jiang原创，但是可以复制、粘贴、转载，因为我们是好朋友！！