其他分享
首页 > 其他分享> > 树套树部分

树套树部分

作者:互联网

样例输入 1

7
add 1 1
add 3 4
find 0 0
remove 1 1
find 0 0
add 1 1
find 0 0

样例输出 1

1 1
3 4
1 1

样例输入 2

13
add 5 5
add 5 6
add 5 7
add 6 5
add 6 6
add 6 7
add 7 5
add 7 6
add 7 7
find 6 6
remove 7 7
find 6 6
find 4 4

样例输出 2

7 7
-1
5 5

数据范围

对于 \(100 \%\) 的数据:\(1 \le n \le 2 \times 10^5,0 \le x,y \le 10^9\)


今天考试的 T2 ,考完感触非常多,写了一个树套树,跑的不如老奶奶,很劣,且正解还有单 set 做法,但还是想讲一讲树套树写法。

题目翻译就是要求一个点 \(i\) 满足:$ min(x < x_i) $ 同时 \(min(y < y_i)\) 。

典型的二维数点,考虑树套树,离散化 \(x\) 轴,对于 \(y\) 轴存入节点的 set 中,针对查询,就是在线段树中查询大于 \(x\) 的最左端点中的 set 值,所以其实只在叶子节点开 set 就可以了。

Code.

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10,imax=INT_MAX;
struct node {int l,r; multiset <int> s; } tr[N<<2];
struct aw {int x,y,op;} q[N];
vector<int> num; int m;
void build(int u,int l,int r)
{
	tr[u].l=l,tr[u].r=r; if(l == r) return ;
	int mid = l + r >> 1;
	build(u<<1,l,mid); build(u<<1|1,mid+1,r);
}
void add(int u,int l,int y)
{
	tr[u].s.insert(y); if(tr[u].l == tr[u].r) return ;
	int mid = tr[u].l + tr[u].r >> 1;
	if(l <= mid) add(u<<1,l,y); else add(u<<1|1,l,y);
}
void remove(int u,int l,int y)
{
	tr[u].s.erase(tr[u].s.find(y)); if(tr[u].l == tr[u].r) return ;
	int mid = tr[u].l + tr[u].r >> 1;
	if(l <= mid) remove(u<<1,l,y); else remove(u<<1|1,l,y);
}
pair <int,int> query(int u,int l,int x)
{
	if(tr[u].s.upper_bound(x) == tr[u].s.end()) return make_pair(imax,imax);
	if(tr[u].l == tr[u].r) return make_pair(tr[u].l,*tr[u].s.upper_bound(x));
	int mid = tr[u].l + tr[u].r >> 1;  pair <int,int> lt=make_pair(imax,imax);
	if(l < mid) lt=query(u<<1,l,x); if(lt.first == imax) lt=query(u<<1|1,l,x);
	return lt;
}
int main()
{
	freopen("shot.in","r",stdin); freopen("shot.out","w",stdout); scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		string a; cin>>a; scanf("%d%d",&q[i].x,&q[i].y); num.push_back(q[i].x);
		if(a == "add") q[i].op=1; else if(a == "remove") q[i].op=2; else q[i].op=3;
	}
	sort(num.begin(),num.end()); num.erase(unique(num.begin(),num.end()),num.end());
	build(1,1,num.size()+1);
	for(int i=1;i<=m;i++)
	{
		int x=lower_bound(num.begin(),num.end(),q[i].x)-num.begin()+1,y=q[i].y;
		if(q[i].op == 1) add(1,x,y);
		else if(q[i].op == 2) remove(1,x,y);
		else 
		{
			auto it=query(1,x,y);
			if(it.second == imax) puts("-1");
			else printf("%d %d\n",num[it.first-1],it.second);
		}
	}
	return 0;
}

标签:树套,int,tr,add,num,imax,部分,find
来源: https://www.cnblogs.com/EastPorridge/p/16534003.html