【线段树】线段树练习题二

Description

桌子上零散地放着若干个不同颜色的盒子，桌子的后方是一堵墙。如下图所示。问从桌子前方可以看到多少个盒子？假设人站得足够远(输入时,由底向上,从左到右)。

Sample Input

16  //桌子长度
5   // 盒子数量
4 7
12 14
1 5
6 10
11 16

Sample Output

4

Hint

1<=n<=100000,1<=m<=100000,保证坐标范围为[1,n].

---

解题思路

将题目转换一下，等于把每个箱子染上不同的颜色（标记），可以看到多少种颜色

---

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,x,y,a[400100],ans,f[100100];
void add(int root,int l,int r){
	if(a[root]>=0)f[a[root]]=1;//桶排记录
	   else {
	   	int mid=(l+r)/2;
	   	add(root*2,l,mid);
	   	add(root*2+1,mid,r);
	   }
}
void insert(int root,int l,int r,int x,int y,int color){
	if(a[root]==color)return;//如果已经是这个颜色就不需要再求下去了（好像没有这个情况qwq）
	if(l==x&&r==y){//找到范围就可以返回了，不需要往下继承
		a[root]=color;
		return;
	}
	if(a[root]>=0){//如果有多种颜色，把颜色往下继承
		a[root*2]=a[root*2+1]=a[root];
		a[root]=-1;//-1表示有多种颜色
	}
	int mid=(l+r)/2;//二分查找
	if(y<=mid)insert(root*2,l,mid,x,y,color);
	   else if(x>=mid)insert(root*2+1,mid,r,x,y,color);
	        else{
	        	insert(root*2,l,mid,x,mid,color);
	        	insert(root*2+1,mid,r,mid,y,color);
	        }
}
int main(){
	scanf("%d%d",&m,&n);
	for(int i=1;i<=n;i++){
		scanf("%d%d",&x,&y);
		insert(1,1,m,x,y,i);
	}
	add(1,1,m);
	for(int i=1;i<=n;i++)
	    if(f[i])ans++;
	printf("%d",ans); 
}