注意:
懒标记中当分裂的时候会进行pushdown,而pushdown中必定会将当前要被分裂的节点的懒标记值清为0,不留保留下来。因为保留下来的话,当下次进行分裂的时候,又会对此节点进行pushdown操作,就会导致错误。
比如,此题flag 先全部被标记为了1, 如果不清0的话,样例中第一步是正确的,使得1~5的区间flag 变为了2, 1~10还是1, 那么当样例第二步操作的时候,1~10再次分裂,1~5的flag又变为了1,导致错误。实际情况应该是1~4为2, 5 为3 , 6 ~9为3,10为1。
所以,切记分裂状况下pushdown操作中,懒标记必须要清空为无影响状态。
题目链接:https://acm.hdu.edu.cn/showproblem.php?pid=1698
题目:
给定一个n, 对于区间[1, n]有如下操作: 选定l, r, 把[l, r]区间全部修改为数值c
最后输出[1, n]的区间和
分析:
首先,根据题意可以知道,求的是1~n的和,也就是直接query()根节点。
因此,需要使用一个sum属性在线段树中,用于维护此节点L ~ R的值的总和。
但是在这个操作之中,还存在一个区间修改, 而区间修改,就会想到需要使用懒标记的方式。而我们线段树中维护的懒标记便是此题中所改变的值c。
在pushdown操作中,我们每将节点的c值往下传给子节点的时候,就将sum值进行更新,随后修改子节点的懒标记值。最后将节点的c值修改为0。
而pushup操作,则通过子节点求和得到根节点的值。
最后使用query()直接查询1~n的总和sum值即可。
代码实现:
# include <iostream>
using namespace std;
const int N = 100010;
int n,q;
int money[] = {0,1,2,3}; //flag为1,铜棒,1元, flag为2,银棒,2元...
struct Node
{
int l,r;
int sum; //为l~r的总值
int flag; //为懒标记,标记它当前的种类
}edgs[N * 4];
void pushup(int u)
{
edgs[u].sum = edgs[2 * u].sum + edgs[2 * u + 1].sum;
}
void pushdown(int u) //猜测修改颜色不需要清空pushdown , 一个与模板的不同之处
{
if(edgs[u].flag != 0)
{
edgs[2 * u].sum =(edgs[2 * u].r - edgs[2 * u].l + 1) * money[edgs[u].flag];
edgs[2 * u].flag = edgs[u].flag;
edgs[2 * u + 1].sum = (edgs[2 * u + 1].r - edgs[2 * u + 1].l + 1) * money[edgs[u].flag];
edgs[2 * u + 1].flag = edgs[u].flag;
edgs[u].flag = 0;
}
}
void build(int u , int l , int r)
{
edgs[u].l = l,edgs[u].r = r;
edgs[u].flag = 1; // 刚开始全部为铜棒
if(l == r)
{
edgs[u].sum = 1;
}
else
{
int mid = (edgs[u].l + edgs[u].r) / 2;
build(2 * u , l , mid);
build(2 * u + 1 ,mid +1 , r);
pushup(u);
}
}
void modify(int u , int l , int r , int v)
{
if(edgs[u].l >= l && edgs[u].r <= r)
{
edgs[u].sum =(edgs[u].r - edgs[u].l + 1) * money[v];
edgs[u].flag = v;
}
else
{
pushdown(u);
int mid = ( edgs[u].l + edgs[u].r ) / 2;
if(l <= mid)
{
modify(2 * u , l , r , v);
}
if(r > mid)
{
modify(2 * u + 1, l , r , v);
}
pushup(u);
}
}
int query(int u , int l , int r)
{
if(edgs[u].l >= l && edgs[u].r <= r)
{
return edgs[u].sum;
}
pushdown(u);
int ans = 0;
int mid = (edgs[u].l + edgs[u].r) / 2;
if(l <= mid)
{
ans += query(2 * u , l , r);
}
if(r > mid)
{
ans += query(2 * u + 1 , l , r);
}
return ans;
}
int main()
{
int loop;
scanf("%d",&loop);
for(int i = 1 ; i <= loop; i++)
{
scanf("%d %d",&n,&q);
build(1,1,n);
while(q--)
{
int x,y,z;
scanf("%d %d %d",&x,&y,&z);
modify(1 , x , y , z);
}
printf("Case %d: The total value of the hook is %d.\n",i,query(1,1,n)); // 直接使用edgs[1].sum实际上也是可以的,两个含义实际上是一样的。都是直接查询根节点的值。
}
return 0;
}