线段树练习四 题解
题目
在平面内有一条长度为n的线段(不计入答案),可以对进行以下2种操作:
1、把从
x
x
x到
y
y
y的再加一条线段
2、查询从
x
x
x到
x
x
x+1有多少条线段
输入
第一行输入
n
n
n,
m
m
m
第2~
m
m
m+1行,每行2个数
x
x
x,
y
y
y,表示从x到y再加一条线段
最后一行输入2个数,为
x
x
x和
x
x
x+1,查询x到x+1的线段数目
输出
输出 x x x到 x x x+1的线段数目
样例
input
7 2
2 5
3 6
4 5
output
2
解题思路
构建一棵树
边输入边覆盖
统计输入的区间的子节点的和
代码
#include<iostream>
#include<cstdio>
using namespace std;
int n,m,x,y;
struct hhx{
int l,r,ans;
}t[401000];
void tree(int d,int l,int r) //建树
{
int mid=(l+r)/2;
t[d].l=l;
t[d].r=r;
if (r-l>1)
tree(d*2,l,mid),tree(d*2+1,mid,r);
}
void in(int d,int l,int r) //覆盖
{
int mid=(t[d].l+t[d].r)/2;
if (t[d].l==l&&t[d].r==r)
t[d].ans++; //线段数+1
else if (r<=mid) in(d*2,l,r);
else if (l>=mid) in(d*2+1,l,r);
else in(d*2,l,mid),in(d*2+1,mid,r);
}
int tj(int d,int l,int r)
{
int ans=t[1].ans;
while (t[d].r-t[d].l>1)
{
int mid=(t[d].l+t[d].r)/2;
if (t[d].l==l&&t[d].r==r) //重合
break;
else if (r<=mid)
d=2*d,ans+=t[d].ans; //向左搜,累加答案
else if (l>=mid)
d=2*d+1,ans+=t[d].ans; //向右搜,累加答案
}
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
tree(1,1,n);
for (int i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
in(1,x,y); //覆盖
}
scanf("%d%d",&x,&y);
printf("%d",tj(1,x,y)); //统计最终答案
}