这个题目的构造方法应该还算是很好想的,先给a按照从小到大排序,然后按顺序插入数据,构造一棵二叉查找树,而且50000的数据,nlogn的做法,应该还是很好的。不过这个题目的编码比想象中要麻烦一点,并且编码完成后居然返回了一个tle。
表示不能理解,生成了一个大数据后,发现如果排序后k也有序,那么就会二叉树退化,n*n的复杂度。简单画画图可以找到一个优化,先找要插入数据的后继或者前驱,然后从哪个位置开始插入,那么可以大大稳定复杂度。找这个东西的工作就交给set来完成即可。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
using namespace std;
const int maxn=5e4+9;
int lon,f[maxn];
struct D
{
int a,k,id;
bool operator <(const D & xx) const
{
return k<xx.k;
}
}data[maxn]; bool cmp(const D &p,const D &q)
{
return p.a<q.a;
} struct
{
int l,r,id,fa;
}tr[maxn]; set <D> s;
int maxm;
void insert(int t)
{
// cout<<t<<endl;
int root;
if(t!=1)
{
if(data[t].k>data[maxm].k)
{
root=f[data[maxm].id];
maxm=t;
}
else root=f[s.lower_bound(data[t])->id];
}
else
{
root=1;
maxm=1;
}
while(tr[root].id!=0)
{
// cout<<root<<endl;
if(data[t].k<data[tr[root].id].k)
{
if(tr[root].l==0)
{
tr[root].l=++lon;
tr[lon].fa=root;
}
root=tr[root].l;
}
else
{
if(tr[root].r==0)
{
tr[root].r=++lon;
tr[lon].fa=root;
}
root=tr[root].r;
}
}
tr[root].id=t;
f[data[t].id]=root;
s.insert(data[t]);
} int main()
{
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
{
scanf("%d %d",&data[i].k,&data[i].a);
data[i].id=i;
}
sort(data+1,data+1+n,cmp);
memset(tr,0,sizeof(tr));
lon=1;
for(int i=1;i<=n;i++)
insert(i);
printf("YES\n");
for(int i=1;i<=n;i++)
{
printf("%d %d %d\n",data[tr[tr[f[i]].fa].id].id,data[tr[tr[f[i]].l].id].id,data[tr[tr[f[i]].r].id].id);
}
}
return 0;
}