codeforces 915E - Physical Education Lessons(线段树动态开点板子题)

题意:

从现在到学期结束还有 n 天(从 1 到 n 编号),
他们一开始都是1。接下来学校的工作人员会依次发出 q 个指令,
每个指令可以用三个参数 l,r,k 描述:
如果 k=1那么从 l 到 r (包含端点)的所有日子都变成0。
如果 k=2那么从 l 到 r (包含端点)的所有日子都变成1。
输出 q 行,第 i 行表示第 i 个指令被下发后1~n的和。
 

思路:线段树动态开点:

#include<bits/stdc++.h>
#define M 300010
using namespace std;
int n,m,tot=1;
struct node
{
    int l,r,sum,lazy;
} T[50*M];
void down(int rt,int l,int r)
{
    if(!T[rt].l) T[rt].l=++tot;//如果没有儿子,开辟新的节点当儿子
    if(!T[rt].r) T[rt].r=++tot;
    int mid=(l+r)>>1;
    T[T[rt].l].sum=(mid-l+1)*T[rt].lazy;
    T[T[rt].l].lazy=T[rt].lazy;
    T[T[rt].r].sum=(r-mid)*T[rt].lazy;
    T[T[rt].r].lazy=T[rt].lazy;
    T[rt].lazy=-1;
}
void update(int rt,int x,int y,int l,int r,int flag)
{
    if(l>=x&&r<=y)
    {
        T[rt].sum=(r-l+1)*flag;
        T[rt].lazy=flag;
        return ;
    }
    if(T[rt].lazy!=-1)down(rt,l,r);//0也要向下传,1也要向下传
    int mid=(l+r)>>1;
    if(x<=mid)update(T[rt].l,x,y,l,mid,flag);
    if(y>mid)update(T[rt].r,x,y,mid+1,r,flag);
    T[rt].sum=T[T[rt].l].sum+T[T[rt].r].sum;
}
int main()
{
    scanf("%d%d",&n,&m);
    int l,r,opt;
    T[1].sum=n,T[1].lazy=1;//初始化根
    for(int i=1; i<=m; i++)
    {
        scanf("%d%d%d",&l,&r,&opt);
        if(opt==1) update(1,l,r,1,n,0);
        if(opt==2) update(1,l,r,1,n,1);
        printf("%d\n",T[1].sum);
    }
}

 

上一篇:字符串算法(string_algorithm)


下一篇:闭包(closure)