校门外的树题解

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

某校大门外长度为L的马路上有一排树，每两棵相邻的树之间的间隔都是1米。我们可以把马路看成一个数轴，马路的一端在数轴0的位置，另一端在L的位置；数轴上的每个整数点，即0，1，2，……，L，都种有一棵树。

由于马路上有一些区域要用来建地铁。这些区域用它们在数轴上的起始点和终止点表示。已知任一区域的起始点和终止点的坐标都是整数，区域之间可能有重合的部分。现在要把这些区域中的树（包括区域端点处的两棵树）移走。你的任务是计算将这些树都移走后，马路上还有多少棵树。

输入描述:

第一行有两个整数：L（1 <= L <= 10000）和 M（1 <= M <= 100），L代表马路的长度，M代表区域的数目，L和M之间用一个空格隔开。接下来的M行每行包含两个不同的整数，用一个空格隔开，表示一个区域的起始点和终止点的坐标。

输出描述:

包括一行，这一行只包含一个整数，表示马路上剩余的树的数目。

示例1

输入

复制500 3 150 300 100 200 470 471

500 3
150 300
100 200
470 471

输出

复制298

298

备注:

对于20%的数据，区域之间没有重合的部分；
对于其它的数据，区域之间有重合的情况。

暴力做法：创建数组a，代表所有的0~l的树，需保留的数标记为1，不需要保留的数标记为0，遍历要砍的区间，遍历完后计算0~l有多少颗树保留，时间复杂度o（n*m）

差分做法：

前置知识：前缀和，差分

推荐视频：https://www.bilibili.com/video/BV17f4y1r78t/?spm_id_from=333.999.0.0&vd_source=c858426e961609aa32b2598376e96f5f

思路：首先对区间数组的修改操作，我们可以联想到利用差分来进行操作来优化对区间数组修改操作的时间复杂度，创建数组a，代表所有的0~l的树，需保留的数标记为1，不需要保留的数为小于1的树。创建差分数组b。需要砍的树的区间全部进行减1操作。最终通过b差分数组还原修改后的a数组，还原后的数组任然为1证明没有被砍。

代码：

#include <iostream>
using namespace std;
int a[10009]={0},b[10009]={0};
int main() {
   int l,m;
    cin>>l>>m;
    for(int i=0;i<=l;i++)a[i]=1;
    b[0]=1;//优化创建差分数组
    for(int i=1;i<=m;i++)
    {
        int temp1,temp2;
        cin>>temp1>>temp2;
        b[temp1]-=1;
        b[temp2+1]+=1;
    }
    int sum=0;
    int count=0;
    for(int i=0;i<=l;i++)
    {
        sum+=b[i];//还原修改后的a[i]
        if(sum>0)count++;
    }
    cout<<count<<endl;
}
// 64 位输出请用 printf("%lld")