树套树部分
作者:互联网
样例输入 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