线段树系列-hdu-1698-Just a Hook-区间修改求总和

#include<bits/stdc++.h>
using namespace std;
struct seg{
int l, r, v;               // v的值如果是-1则说明这条线段不是单色
}tree[410000];
void build(int l, int r, int k) {
tree[k].l = l;
tree[k].r = r;
tree[k].v = 1;
if(l == r) return;
int mid = (l + r) >> 1;
build(l, mid, 2*k);
build(mid+1 , r, 2*k+1);
}
void update(int l, int r, int v, int k) {
if(tree[k].v == v) return;                 // 如果颜色相同就直接返回
if(tree[k].l == l && tree[k].r == r) {     // 如果就是目标区域就直接修改线段的值
tree[k].v = v;
return;
}
if(tree[k].v != -1) {                      //如果是纯色，因为修改后不是纯色，所以要继续修改子区间，
tree[2*k].v = tree[k].v;               //所以子区间的值先赋值为原区间的值，原区间设置为杂色
tree[2*k+1].v = tree[k].v;             //为后面的修改做准备
tree[k].v = -1;
}
int mid = (tree[k].l + tree[k].r) >> 1;
if(r <= mid) update(l, r, v, 2*k);
else if(l > mid) update(l, r, v, 2*k+1);
else {
update(l, mid, v, 2*k);
update(mid+1, r, v, 2*k+1);
}

}
int query(int k) {
if(tree[k].v != -1) {
return tree[k].v * (tree[k].r - tree[k].l + 1);
}
else {
return query(k*2) + query(k*2+1);
}
}
int main()
{
int t;
int n, q;
int x, y, v;
int kase = 0;
scanf("%d", &t);
while(t--) {
scanf("%d", &n);
build(1, n, 1);
scanf("%d", &q);
for(int i = 1; i <= q; i++) {
scanf("%d%d%d", &x, &y, &v);
update(x, y, v, 1);
}
printf("Case %d: The total value of the hook is %d.\n", ++kase, query(1));
}
return 0;
}