首页 > 代码库 > 【BZOJ】2648: SJY摆棋子 & 2716: [Violet 3]天使玩偶（kdtree）
【BZOJ】2648: SJY摆棋子 & 2716: [Violet 3]天使玩偶（kdtree）
http://www.lydsy.com/JudgeOnline/problem.php?id=2716
http://www.lydsy.com/JudgeOnline/problem.php?id=2648
双倍经验题。。。
kdtree裸题吧。。。。。今天学了下kdtree。。。感觉挺简单的。。。。
其实就是对几何图形进行剖分建树，而特殊在于，x和y轴轮流剖。。。。这样可以提供很好的性质以便于查找。。。
（一开始脑补了个treap代替kdtree。。。。。显然我脑残了。。。。写到一半发现这是不能旋转的。。。。。
（然后又脑补了下。。。。这货如果和孙子（儿子的儿子）可以旋转。。。。。。。。因为这货的节点类似红黑树。。。但是蒟蒻我太弱从来没写过红黑树。。。
很好的资料（概念脑补）：http://www.cnblogs.com/v-July-v/archive/2012/11/20/3125419.html
一些参考代码（虽然说完全没按照这样写）：http://blog.csdn.net/jiangshibiao/article/details/34144829
说一下怎么查找。
首先这是曼哈顿距离。。和欧几里得距离不同。。。如果欧几里得那么简单多了。。
我们需要维护极值（即能包围所有子树包括自己的点的一个矩形的四个顶点）：
如果要查找的点在矩形内，那么进入查找。
如果在矩形外，那么计算出查找点到矩形的曼哈顿距离，判断是否小于当前最优解，如果是，进入搜索。
（或许还能优化，因为一开始我的想法和这个不怎么一样。。我觉得。。递归进入两棵子树中其中一棵后（按x或y排序而不是找极值），然后判断最优解是否能从左子树的某个点越过当前根的分割线。。。。。如果是，就进入。。。
然后就没了。。。
留下的坑：欧几里得的没写过。。。删除操作没写过（觉对要斯巴达。。。。想写成平衡树（强迫症系列
#include <cstdio>#include <cstring>#include <cmath>#include <string>#include <iostream>#include <algorithm>#include <queue>#include <set>#include <map>using namespace std;typedef long long ll;#define rep(i, n) for(int i=0; i<(n); ++i)#define for1(i,a,n) for(int i=(a);i<=(n);++i)#define for2(i,a,n) for(int i=(a);i<(n);++i)#define for3(i,a,n) for(int i=(a);i>=(n);--i)#define for4(i,a,n) for(int i=(a);i>(n);--i)#define CC(i,a) memset(i,a,sizeof(i))#define read(a) a=getint()#define print(a) printf("%d", a)#define dbg(x) cout << (#x) << " = " << (x) << endl#define error(x) (!(x)?puts("error"):0)#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }const int N=2000005, oo=~0u>>1;struct node *null;struct node { node *c[2]; int mx[2], mn[2], p[2]; void upd(node *x) { if(x==null) return; mx[0]=max(mx[0], x->mx[0]); mx[1]=max(mx[1], x->mx[1]); mn[0]=min(mn[0], x->mn[0]); mn[1]=min(mn[1], x->mn[1]); } int getdis(node *x) { int ret=0; ret+=max(x->p[0]-mx[0], 0); ret+=max(x->p[1]-mx[1], 0); ret+=max(mn[0]-x->p[0], 0); ret+=max(mn[1]-x->p[1], 0); return ret; } void init(int x, int y) { p[0]=mx[0]=mn[0]=x; p[1]=mx[1]=mn[1]=y; c[0]=c[1]=null; } int dis(node *x) { return abs(p[0]-x->p[0])+abs(p[1]-x->p[1]); }}*root, T[N], *Tcnt=T;node *newnode(int x=0, int y=0) { Tcnt->init(x, y); return Tcnt++; }void init() { null=newnode(); null->c[0]=null->c[1]=null; root=null; }void insert(node *&x, node *y, bool f) { if(x==null) { x=y; return; } bool d=x->p[f]<y->p[f]; x->upd(y); insert(x->c[d], y, !f);}void ins(int x, int y) { insert(root, newnode(x, y), 0); }void ask(node *x, node *y, int &ans) { ans=min(ans, x->dis(y)); int d[2]; d[0]=x->c[0]==null?oo:x->c[0]->getdis(y); d[1]=x->c[1]==null?oo:x->c[1]->getdis(y); bool f=d[0]>d[1]; if(d[f]==oo) return; ask(x->c[f], y, ans); if(d[!f]<ans) ask(x->c[!f], y, ans);}int ask(int x, int y) { int ret=oo; static node r; r.p[0]=x; r.p[1]=y; ask(root, &r, ret); return ret;}int main() { init(); int n=getint(), m=getint(); rep(i, n) { int x=getint(), y=getint(); ins(x, y); } while(m--) { int t=getint(), x=getint(), y=getint(); if(t==2) printf("%d\n", ask(x, y)); else ins(x, y); } return 0;}
Description
Input
Output
Sample Input
1 1
2 3
2 1 2
1 3 3
2 4 2
Sample Output
1
2
HINT
kdtree可以过
Source
鸣谢 孙嘉裕
【BZOJ】2648: SJY摆棋子 & 2716: [Violet 3]天使玩偶（kdtree）