新手写的线段树,看了定义就写,代码风格不好,请大神指教,题目意思就是简单维护区间和。。。第一篇博客,感觉有必要总结经验。。`
#include<iostream>
using namespace std;
const int size1 = 400000;//大概开到4倍
int shu[size1];
int lazy[size1];//懒惰标记,即是需要在对这区间的儿子们进行操作
void build(int l, int r, int k) //建树,通过递归去建立
{
if (l == r) //L,R是区间,很容易混淆,k只是数组存树的一个映射。所以下面基本都是折中操作,
{ //否则K会出错,这里是到了叶子节点,就是我们的单个参数,自然建立后跳出
shu[k] = 1;
return;
}
int mid = (l + r) / 2;
build(l, mid, k * 2);//建立左子树
build(mid + 1, r, k * 2 + 1);//建立右子树
shu[k] = shu[2 * k + 1] + shu[2 * k]; //这里是维护线段树的保存信息,通过递归从底而上维护,从而保证了正确性
}
void change(int l, int r, int k, int num)
{
if (l == r) //如果是叶子,懒惰标记到此为止
{
lazy[k] = 0;
}
else
lazy[k] = num;//懒惰标记需要下放
shu[k] = (r - l + 1)*num;//维护区间和
}
void change(int l, int r, int k, int x, int y, int num)//修改函数,由于第一次写,按照定义也没参考别人优秀代码,思维混乱导致代码不好。
{
if (l == r)//叶子,修改后结束,等待自动回溯更新父节点信息
{
shu[k] = num;
return;
}
else//若不是叶子
{
if (l != r && (l <= x || r >= y) && lazy[k] != 0) //如果懒惰标记存在,且在操作区间内
{
int mid = (l + r) / 2;
change(l, mid, k * 2, lazy[k]);//下放懒惰标记的操作到儿子
change(mid + 1, r, k * 2 + 1, lazy[k]);
lazy[k] = 0;//取消懒惰标记
}
if (l == x&&r == y&&l != r || (x <= l&&r <= y))//如果操作的区间刚好为递归区间或者递归区间被操作区间包涵,下放操作到节点儿子
{
lazy[k] = num;
shu[k] = (r - l + 1)*num;
return;
}
else//否则,递归遍历左右儿子
{
int mid = (l + r) / 2;
if (x <= mid)//逼近原理,即如果操作区间在递归区间的中间的左边,则递归处理左儿子,这样肯定可以刚好递归到操作区间
{
change(l, mid, k * 2, x, y, num);
}
if (y>mid)//同上的右儿子版本
{
change(mid + 1, r, k * 2 + 1, x, y, num);
}
}
}
shu[k] = shu[2 * k + 1] + shu[2 * k];//回溯维护父节点信息
}
int main()
{
int t;
cin >> t;
for (int i = 1; i <= t; i++)
{
memset(lazy, 0, sizeof(lazy));
memset(shu, 0, sizeof(shu));
int n;
cin >> n;
build(1, n, 1);
int q;
cin >> q;
for (int i = 0; i<q; i++)
{
int x, y, num;
scanf("%d%d%d", &x, &y, &num);
change(1, n, 1, x, y, num);
}
printf("%s%d%c%s%d%c\n", "Case ", i, ':', " The total value of the hook is ", shu[1],'.');//因为问整段区间和,访问根节点就好
}
}