洛谷 P1276 校门外的树（线段树）

P1276 校门外的树（增强版）

题目描述

校门外马路上本来从编号0到L，每一编号的位置都有1棵树。有砍树者每次从编号A到B处连续砍掉每1棵树，就连树苗也不放过（记 0 A B ，含A和B）；幸运的是还有植树者每次从编号C到D 中凡是空穴（树被砍且还没种上树苗或树苗又被砍掉）的地方都补种上树苗（记 1 C D，含C和D）；问最终校门外留下的树苗多少棵？植树者种上又被砍掉的树苗有多少棵？

输入输出格式

输入格式：
第一行L和N,表示校园外原来有L+1棵树，并有N次砍树或种树的操作。

以下N行，砍树或植树的标记和范围，每行3个整数。

L（1 <= L <= 10000）和 N（1 <= N <= 100）

输出格式：
共两行。第1行校门外留下的树苗数目，第2行种上又被拔掉的树苗数目。

输入输出样例

输入样例#1：
10 3
0 2 6
1 1 8
0 5 7
输出样例#1：
3
2

分析：记录每个区间状态，如果全是树记为1，全是树苗记为2，全是空穴记为0，否则记为-1，注意问题即可。

代码

#include <cstdio>
#define N 100000
using namespace std;

struct tnode
{
    int l,r,sta;
}tr[N];
int l,n,a1,a2;

void build(int p)
{
    if (tr[p].l==tr[p].r)
    {
        tr[p].sta=1;
        return;
    }
    int mid=(tr[p].l+tr[p].r)/2;
    tr[p*2].l=tr[p].l;
    tr[p*2].r=mid;
    tr[p*2+1].l=mid+1;
    tr[p*2+1].r=tr[p].r;
    tr[p*2].sta=tr[p*2+1].sta=1;
    build(p*2);
    build(p*2+1);
}

void cut(int p,int x,int y)
{
    if (tr[p].sta!=-1) tr[p*2].sta=tr[p*2+1].sta=tr[p].sta;
    if (tr[p].sta==0) return;
    if (tr[p].l==x&&tr[p].r==y&&tr[p].sta!=-1)
    {
        if (tr[p].sta==2) 
        {
            a2+=tr[p].r-tr[p].l+1;
            a1-=tr[p].r-tr[p].l+1;
        }
        tr[p].sta=0;
        return;
    }
    int mid=(tr[p].l+tr[p].r)/2;
    if (y<=mid) cut(p*2,x,y);
        else if (x>mid) cut(p*2+1,x,y);
            else cut(p*2,x,mid),cut(p*2+1,mid+1,y);
    if (tr[p*2].sta!=tr[p*2+1].sta) tr[p].sta=-1;
        else tr[p].sta=tr[p*2].sta;
}

void plant(int p,int x,int y)
{
    if (tr[p].sta!=-1) tr[p*2].sta=tr[p*2+1].sta=tr[p].sta;
    if (tr[p].sta==1||tr[p].sta==2) return;
    if (tr[p].l==x&&tr[p].r==y&&tr[p].sta!=-1)
    {
        tr[p].sta=2;
        a1+=tr[p].r-tr[p].l+1;
        return;
    }
    int mid=(tr[p].l+tr[p].r)/2;
    if (y<=mid) plant(p*2,x,y);
        else if (x>mid) plant(p*2+1,x,y);
            else plant(p*2,x,mid),plant(p*2+1,mid+1,y);
    if (tr[p*2].sta!=tr[p*2+1].sta) tr[p].sta=-1;
        else tr[p].sta=tr[p*2].sta;
}

int main()
{
    scanf("%d%d",&l,&n);
    tr[1].l=0;
    tr[1].r=l;
    tr[1].sta=1;
    build(1);
    for (int i=1;i<=n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&z,&x,&y);
        if (z) plant(1,x,y);
            else cut(1,x,y);
    }
    printf("%d\n%d",a1,a2);
}