计蒜客:帕吉的肉钩-区间更新-区间和

帕吉的肉钩

在DotA游戏中,帕吉的肉钩是很多英雄最害怕的东西。钩子由连续若干段的等长金属棒制成。
现在帕吉对钩子由一些操作:
我们将金属棒1~n依次编号,帕吉可以把编号x~y的金属棒变成铜棒、银棒、金棒。
每段铜棒的价值是1;每段银棒的价值是2;每段金棒的价值是3。
肉钩的总价值是n段金属棒价值之和。帕吉想知道若干操作以后钩子的总价值。
输入格式
第一行一个整数N(1≤N≤105),表示肉钩金属棒的数量,初始状态为铜棒。
第二行一个整数Q(1≤Q≤105),表示操作数。
接下来Q行,一行三个整数,X,Y,Z(1≤X≤Y≤N,1≤Z≤3)。
输出格式
一行一个整数,表示钩子的总价值,用样例中的格式。

代码:

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

int n,q;
const int MAX_N = 1e5+5;
int s[4 * MAX_N],col[4 * MAX_N];

//父节点合并 
void up(int p){
    s[p] = s[p * 2] + s[p * 2 + 1];
}

//向下更新一层  区间更新使用延迟更新的思想提高效率
void down(int p, int l, int r){
    if(col[p]){
        int mid = (l + r) / 2;
        s[p * 2] = col[p] * (mid - l + 1);
        s[p * 2 + 1] = col[p] * (r - mid);
        col[p * 2] = col[p];
        col[p * 2 + 1] = col[p];
        col[p] = 0;
    }
}

//更新 
void modify(int p,int l,int r,int x,int y,int v){
    if(x <= l && r <=y){
		s[p] = (r - l + 1) * v; //对父结点修改
        col[p] = v; //标记子节点需要延迟更新 
        return;
    }
    down(p, l, r);//向下更新一层 
    int mid = (l + r)/2;
    if(x <= mid){
        modify(p * 2, l, mid, x, y, v);
    }
    if(y > mid){
        modify(p * 2 + 1, mid + 1,r,x,y,v);
    }
    up(p);
}

//查询 
int query(int p,int l,int r,int x,int y){
	if(x <= l && r <= y){
        return s[p];
    }
    down(p,l,r);//如果没有return,这时子节点必须更新了  就向下更新一层
    int mid = (l + r) / 2, res = 0;
    if(x <= mid){
		res += query(p * 2, l,mid,x,y);
    }
    if(y > mid){
        res += query(p * 2 + 1,mid + 1,r,x,y);
    }
    return res;
}

int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++){
		modify(1,1,n,i,i,1);
	}
	while(q--){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		modify(1,1,n,x,y,z);
	}
	cout<<"The total value of the hook is "<<query(1,1,n,1,n)<<"."<<endl; 
	return 0;
}
原文地址:https://www.cnblogs.com/fisherss/p/10369808.html