Codeup 问题 A: 剩下的树 （算法笔记课后习题）

Codeup 问题 A: 剩下的树

题目链接：点击这里

想法：

1. 有一种方法是直接定义很大的bool型数组，用遍历的方式改变true/false来确定某位置是否有树，最后再遍历计数，剩下多少；本文用另一种方法
2. 用合并数组的方法，取数组的并集，使得每个数组不相交，加减后得答案

步骤

1. 接收数据存至二维数组arr（vector< vector >），以二维数组arr的第一列元素为准，对整个二维数组arr进行排序（升序）
2. 进行合并，合并后存到二维数组merge中，开始遍历arr
   （1）若merge为空，或者当前区间的左端比merge最后一个区间的右端要大，这两个区间无交集，直接插入到merge
   （2）否则这两个区间重合，修改merge最后一个区间的右端，取两者右端的最大值
3. 用L+1 减去（各个区间长度+1）即可

#include <cstdio>
#include <vector>
#include <algorithm>

using namespace std;

static bool cmp(const vector<int> &a,const vector<int> &b){
	return a.front()<b.front();
}

int main()
{
	int L,M;

	while(scanf("%d%d",&L,&M)&&L&&M){
		vector<vector<int> > arr;
		vector<vector<int> > merge; //所有的不相交的区间
		arr.clear();
		merge.clear();
		for(int i=0;i<M;i++){
			int l,r;
			scanf("%d%d",&l,&r);
			arr.push_back({l,r});
		}
		sort(arr.begin(),arr.end(),cmp);

		for(int i=0;i<arr.size();i++){
			int l=arr[i][0],r=arr[i][1];
			if(!merge.size() || l>merge.back()[1]){
				merge.push_back({l,r});
			}else{
				merge.back()[1]= (merge.back()[1]>r)? merge.back()[1]:r;
			}
	
		}
		
		for(int i=0;i<merge.size();i++){
			L-=(merge[i][1]-merge[i][0]+1);
		}
		
		printf("%d\n",L+1);
	}
		
	return 0;
}

ps：如果有问题，欢迎交流