原题链接: AcWing. 422 校门外的树
关键词: 区间合并
某校大门外长度为 L 的马路上有一排树,每两棵相邻的树之间的间隔都是 1 米。
我们可以把马路看成一个数轴,马路的一端在数轴 0 的位置,另一端在 L 的位置;数轴上的每个整数点,即 0,1,2,……,L,都种有一棵树。
由于马路上有一些区域要用来建地铁。
这些区域用它们在数轴上的起始点和终止点表示。
已知任一区域的起始点和终止点的坐标都是整数,区域之间可能有重合的部分。
现在要把这些区域中的树(包括区域端点处的两棵树)移走。
你的任务是计算将这些树都移走后,马路上还有多少棵树。
输入格式
输入文件的第一行有两个整数 L 和 M,L 代表马路的长度,M 代表区域的数目,L 和 M 之间用一个空格隔开。
接下来的 M 行每行包含两个不同的整数,用一个空格隔开,表示一个区域的起始点和终止点的坐标。
输出格式
输出文件包括一行,这一行只包含一个整数,表示马路上剩余的树的数目。
数据范围
1≤L≤10000,
1≤M≤100
输入样例:
500 3
150 300
100 200
470 471
输出样例:
298
时/空限制: 1s / 128MB
做法一: 刷数组
- 开一个bool数组,用于记录每棵树的状态,true代表存在,false代表不存在
- 读入每次砍树的区间,将区间内的树的状态赋值为false,最后用循环来统计剩余的树的数目
- 最多100的区间,每个区间长度最多是105,总共就是108,时间在1s内因此是可以通过的
- 算法的时间复杂度大概是O(lm)
代码实现:
#include <iostream>
using namespace std;
int l, m, cnt = 0;
const int maxn = 10010;
bool tree[maxn];
int main(){
scanf("%d %d", &l, &m);
for(int i = 0; i <= l; i ++ ) tree[i] = true;
for(int i = 0; i < m; i ++ ){
int a, b; //区间[a,b]
scanf("%d %d", &a, &b);
for(int j = a; j <= b; j ++ ){
tree[j] = false;
}
}
for(int i = 0; i <= l; i ++ ){
if(tree[i] == true) cnt++;
}
printf("%d", cnt);
return 0;
}
做法二: 区间合并
首先将所有区间按照左端点大小排序;
从左到右遍历每个区间,当前区间为[L, R],下一个区间为[st, ed]
- 如果st < R,说明区间有交集需要合并区间:R = max(R, ed)
- 如果st > R,说明区间没交集。则将当前的[L, R]保存下来,然后[L, R] 赋值为[st, ed],因为区间是按左端点从小到大排序的,所以后面的区间一定不会和当前区间有交集
时间复杂度:O(nlogn)
用结构体数组来存区间:
#include <iostream>
#include <algorithm>
using namespace std;
int m, n; //m马路长度 n区间个数
const int maxn = 110;
struct segment{
int l, r; //区间左右端点
//重载小于号,按照l的值从小到大排序
bool operator< (const segment& t) const{
return l < t.l;
}
}seg[maxn];
int main(){
cin >> m >> n;
for(int i = 0; i < n; i ++ ) cin >> seg[i].l >> seg[i].r;
sort(seg, seg+n);
int sum = 0;
int L = seg[0].l, R = seg[0].r;
for(int i = 1; i < n; i ++ ){
if(seg[i].l <= R) R = max(R, seg[i].r);
else{
sum += R - L + 1;
L = seg[i].l, R = seg[i].r;
}
}
sum += R - L + 1;
cout << m + 1 - sum << endl;
return 0;
}
用pair来存区间:
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int, int> PII;
#define x first
#define y second
int m, n; //m马路长度 n区间个数
const int maxn = 110;
PII q[maxn];
int main(){
cin >> m >> n;
for(int i = 0; i < n; i ++ ) cin >> q[i].x >> q[i].y;
sort(q, q+n);
int sum = 0;
int L = q[0].x, R = q[0].y;
for(int i = 1; i < n; i ++ ){
if(q[i].x <= R) R = max(R, q[i].y);
else{
sum += R - L + 1;
L = q[i].x, R = q[i].y;
}
}
sum += R - L + 1;
cout << m + 1 - sum << endl;
return 0;
}