洛谷 1276 校门外的树（增强版）

Posted 2020-10-31 Driver_Lao

【题解】

　　用线段树维护即可。区间设为一个数以及询问区间的和。第二问的答案可以用砍掉的树的总数减去砍树区间覆盖的范围。

　　

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define N 200010
 4 #define rg register
 5 #define ls (u<<1)
 6 #define rs (u<<1|1)
 7 using namespace std;
 8 int n,m,cnt,tot,s[N],ans;
 9 struct tree{
10     int l,r,sum,set,num;
11 }a[N];
12 inline int read(){
13     int k=0,f=1; char c=getchar();
14     while(c<\'0\'||c>\'9\')c==\'-\'&&(f=-1),c=getchar();
15     while(\'0\'<=c&&c<=\'9\')k=k*10+c-\'0\',c=getchar();
16     return k*f;
17 }
18 inline void pushdown(int u){
19     if(!a[u].set) return;
20     a[ls].set=a[rs].set=1;
21     a[ls].num=a[rs].num=a[u].num;
22     a[ls].sum=a[u].num*(a[ls].r-a[ls].l+1);
23     a[rs].sum=a[u].num*(a[rs].r-a[rs].l+1);
24     a[u].set=0;
25 }
26 void build(int u,int l,int r){
27     a[u].l=l; a[u].r=r;
28     if(l<r){
29         int mid=(l+r)>>1;
30         build(ls,l,mid); build(rs,mid+1,r);
31         a[u].sum=a[ls].sum+a[rs].sum;
32     }
33     else a[u].sum=1;
34 }
35 void update(int u,int l,int r,int s){
36     if(l<=a[u].l&&a[u].r<=r){
37         a[u].set=1; a[u].num=s; a[u].sum=s*(a[u].r-a[u].l+1);
38         return;
39     }
40     pushdown(u); int mid=(a[u].l+a[u].r)>>1;
41     if(l<=mid) update(ls,l,r,s);
42     if(r>mid) update(rs,l,r,s);
43     a[u].sum=a[ls].sum+a[rs].sum;
44 }
45 int query(int u,int l,int r){
46     if(l<=a[u].l&&a[u].r<=r) return a[u].sum;
47     pushdown(u); int mid=(a[u].l+a[u].r)>>1,ret=0;
48     if(l<=mid) ret+=query(ls,l,r);
49     if(r>mid) ret+=query(rs,l,r);
50     return ret;
51 }
52 int main(){
53     n=read(); m=read(); build(1,0,n);
54     for(rg int i=1;i<=m;i++){
55         int opt=read(),l=read(),r=read();
56         if(!opt) cnt+=query(1,l,r),update(1,l,r,0),s[l]++,s[r+1]--;
57         else update(1,l,r,1);
58     }
59     if(s[0]) tot++,ans+=query(1,1,1);
60     for(rg int i=1;i<=n;i++){
61         s[i]+=s[i-1];
62         if(s[i]>0) tot++,ans+=query(1,i,i);
63     }
64     printf("%d\\n",ans);
65     printf("%d\\n",cnt-tot);
66     return 0;
67 }

View Code