【CodeForces】915 E. Physical Education Lessons 线段树

【题目】E. Physical Education Lessons

【题意】10^9范围的区间覆盖，至多3*10^5次区间询问。

【算法】线段树

【题解】每次询问至多增加两段区间，提前括号分段后线段树。

#include<cstdio>
#include<cctype>
#include<set>
#include<algorithm>
using namespace std;
int read(){
    char c;int s=,t=;
    while(!isdigit(c=getchar()))if(c=='-')t=-;
    do{s=s*+c-'';}while(isdigit(c=getchar()));
    return s*t;
}
const int maxn=;
struct tree{int l,r,delta,sum,p;}t[maxn*];//
int au[maxn],av[maxn],k[maxn],a[maxn],n,q,tot;
set<int>s;
void build(int k,int l,int r){
    t[k].l=l;t[k].r=r;t[k].delta=-;
    if(l==r){t[k].p=t[k].sum=a[l]-a[l-];return;}
    int mid=(l+r)>>;
    build(k<<,l,mid);build(k<<|,mid+,r);
    t[k].p=t[k<<].p+t[k<<|].p;
    t[k].sum=t[k<<].sum+t[k<<|].sum;
}
void modify(int k,int x){t[k].sum=t[k].p*x;t[k].delta=x;}
void down(int k){
    if(~t[k].delta){
        modify(k<<,t[k].delta);modify(k<<|,t[k].delta);
        t[k].delta=-;
    }
}
void up(int k){t[k].sum=t[k<<].sum+t[k<<|].sum;}
void cover(int k,int l,int r,int x){
    if(l<=t[k].l&&t[k].r<=r){modify(k,x);return;}
    down(k);
    int mid=(t[k].l+t[k].r)>>;
    if(l<=mid)cover(k<<,l,r,x);
    if(r>mid)cover(k<<|,l,r,x);
    up(k);
}
int main(){
    n=read();q=read();
    for(int i=;i<=q;i++){
        au[i]=read();av[i]=read();k[i]=read();
        s.insert(au[i]-);s.insert(av[i]);
    }
    if(*s.begin()==)s.erase(s.begin());
    s.insert(n);
    for(set<int>::iterator it=s.begin();it!=s.end();it++){
        a[++tot]=*it;
    }
    for(int i=;i<=q;i++){
        au[i]=lower_bound(a+,a+tot+,au[i])-a;
        av[i]=lower_bound(a+,a+tot+,av[i])-a;
    }
    n=tot;
    build(,,n);
    for(int i=;i<=q;i++){
        cover(,au[i],av[i],k[i]-);
        printf("%d\n",t[].sum);
    }
    return ;
}