【SSL 2647】线段树练习题四【线段树变式】

线段树练习四

Time Limit:10000MS Memory Limit:65536K
Case Time Limit:1000MS

Description

在平面内有一条长度为n的线段(不计入答案),可以对进行以下2种操作:
1、把从x到y的再加一条线段
2、查询从x到x+1有多少条线段

Input

第一行输入n,m
第2~m+1行,每行2个数x,y,表示从x到y再加一条线段
最后一行输入2个数,为x和x+1,查询x到x+1的线段数目

Output

输出x到x+1的线段数目

Sample Input

7 2
2 5
3 6
4 5

Sample Output

2

分析:

要建一棵 统计单位区间上所有重合的线段数 就要叶节点到根节点全部累加
也就是:
当前节点区间里线段的数量+这个节点的父节点区间里线段的数量+这个节点的父节点的父节点区间里线段的数量+……+根节点区间里线段的数量=这个节点表示的区间有多少条线段

CODE:

#include
#include
#include
using namespace std;
int n,l,ans,x,y,s,e;
struct node{
	int x,y,count;
}a[400001];
void insert(int dep,int x,int y){  //插入函数
	int mid=(a[dep].x+a[dep].y)>>1;
	if(a[dep].x==x&&a[dep].y==y)  //覆盖
	{
		a[dep].count++;return;  //计数
	}
	else if(y<=mid) insert(2*dep,x,y);  //在左孩子
	else if(x>=mid) insert(2*dep+1,x,y);  //在右孩子
	else{
		insert(2*dep,x,mid);
		insert(2*dep+1,mid,y);  //各有部分(模板)
	}
}
void build(int i){
	if(a[i].y-a[i].x>1){
		int mid=(a[i].x+a[i].y)>>1;
		a[i*2].x=a[i].x;a[i*2].y=mid;
		a[i*2+1].x=mid;a[i*2+1].y=a[i].y;
		build(i*2);build(i*2+1);  //建树
	}
}
int cnt(int dep,int l,int r){  //统计
	ans=a[1].count;
	while(a[dep].y-a[dep].x>1)
	{
		int mid=(a[dep].y+a[dep].x)>>1;
		if(l==a[dep].x&&r==a[dep].y)  //如果到当前区间就退出了
			break;
		if(r<=mid)  //在左
		{
			dep*=2;
			ans+=a[dep].count;
		}
		if(l>=mid){  //在右
			dep=dep*2+1;
			ans+=a[dep].count;
		}
	}
	return ans;
}
int main(){
	scanf("%d%d",&l,&n);
	a[1].x=1;a[1].y=l;
	build(1);
	for(int i=1;i<=n;i++){
		scanf("%d%d",&x,&y);
		insert(1,x,y);  //插入线段
	}
	scanf("%d%d",&s,&e);
	cout<<cnt(1,s,e);
	return 0;
}

你可能感兴趣的:(离散化&线段树,SSLOnlineJudge)