【KD-TREE简介】于SYC1999大神“迷住”下一个。我开始接触这样的算法。

首先。这个概念大概能去百度百科。详细的实施。我在看RZZ的代码长大的。

我们能够想象在平面上有N个点。

首先,按横坐标排序找到最中间的那个点。

然后水平划一条线,把平面分成左右两个部分。再递归调用左右两块。

注意。在第二次(偶数次)调用的时候,是找到纵坐标中最中间的点,并垂直画一条线。

这样效率看上去非常好。维护的时候有点像线段树。

每一个点记录它的坐标、它辖管的区间4个方向的极值、它的左右(或上下)的两个点的标号。

递归两个子树时。注意要up更新这个点辖管的范围。

inline int cmp(arr a,arr b){return a.d[D]<b.d[D]||a.d[D]==b.d[D]&&a.d[D^1]<b.d[D^1];}
inline void up(int k,int s)
{
a[k].min[0]=min(a[k].min[0],a[s].min[0]);
a[k].max[0]=max(a[k].max[0],a[s].max[0]);
a[k].min[1]=min(a[k].min[1],a[s].min[1]);
a[k].max[1]=max(a[k].max[1],a[s].max[1]);
}
int build(int l,int r,int dd)
{
D=dd;int mid=(l+r)>>1;
nth_element(a+l+1,a+mid+1,a+r+1,cmp);
a[mid].min[0]=a[mid].max[0]=a[mid].d[0];
a[mid].min[1]=a[mid].max[1]=a[mid].d[1];
if (l!=mid) a[mid].l=build(l,mid-1,dd^1);
if (mid!=r) a[mid].r=build(mid+1,r,dd^1);
if (a[mid].l) up(mid,a[mid].l);
if (a[mid].r) up(mid,a[mid].r);
return mid;
}

介绍一下nth_element这个STL。头文件就是algorithm。它相当于快排的一部分,调用格式如上。

意思是把第MID个数按cmp放在中间。把比mid“小”的数放在左边,否则放在右边。(注意:不保证左边和右边有序)

上述代码非常好理解。

然后先在我要支持增加点。也是类似于线段树的思想:

void insert(int k)
{
int p=root;D=0;
while (orzSYC)
{
up(p,k);
if (a[k].d[D]<=a[p].d[D]){if (!a[p].l) {a[p].l=k;return;} p=a[p].l;}
else {if (!a[p].r) {a[p].r=k;return;} p=a[p].r;}
D^=1;
}
}

为什么我忽然认为是splay的insert操作?就是每次往某个点的左或右(或者上或下)过去。

比方我们要查询与(x,y)近期的点(曼哈顿距离)与其的距离。

int getdis(int k)
{
int res=0;
if (x<a[k].min[0]) res+=a[k].min[0]-x;
if (x>a[k].max[0]) res+=x-a[k].max[0];
if (y<a[k].min[1]) res+=a[k].min[1]-y;
if (y>a[k].max[1]) res+=y-a[k].max[1];
return res;
}
void ask(int k)
{
int d0=abs(a[k].d[0]-x)+abs(a[k].d[1]-y);
if (d0<ans) ans=d0;
int dl=(a[k].l)? getdis(a[k].l):INF;
int dr=(a[k].r)?getdis(a[k].r):INF;
if (dl<dr){if (dl<ans) ask(a[k].l);if (dr<ans) ask(a[k].r);}
else {if (dr<ans) ask(a[k].r);if (dl<ans) ask(a[k].l);}
}

getdis有点像Astar中的“估价函数”。

计算(x。y)与当前点范围的差距有多少,然后按顺序遍历左二子和右儿子。

这样,假设更新到最优值。就能及时退出。这样的算法在随机数据上是lg的。可是在构造数据上约是sqrt的。

【BZOJ2716&2648】双倍经验。就是裸的K-D TREE模板套套。无压力1A~。

【BZOJ3053】哎。说多了都是泪。

这道题调了不知道多少时间。首先,它拓展到了K维空间上。

这样,cmp就仅仅需推断某一位的大小即可了。然后要查询前m优值。由于m<=10。我为了效率,直接一遍做,开了一个数组记录最优值。然后推断最优值的时候裸O(n)(均摊)的更新答案。

对于那个预计函数也要稍稍改一下(由于是欧几里得距离)。怎么方便怎么来!(反正仅仅会影响到效率)

调了半天后。总算小数据对拍没有问题了~~浪交!T了。。

后来我预计在更新答案时速度太慢。于是一咬牙,把10个最优解开成了队列......

然后大数据对拍~~什么,秒WA?这下真的调了一个下午(由于我是刚学的),后来发现:RZZ的博客里的nth过程用错了。比方从l到r,中间是mid(默认数组下标从1開始),应该是a+l,a+mid,a+r+1。

最后一个+1由于是虚指针。

可是前面都不用+1的(上面的代码已经改动过了)!

!!

最后又是RE。果断要数据!——发现仅仅有一个測试点。我先写了个程序。把測试点拆成了好几个。

然后一測:全过。和在一起:RE!

原来,l和r要及时清零!!!呵呵,多么痛的领悟。

【截取程序】

var
ss:string;
cnt,n,m,a,i,j:longint;
begin
assign(input,'T.in');
reset(input);
while (not(eof)) do
begin
inc(cnt);
str(cnt,ss);
ss:='T'+ss+'.in';
assign(output,ss);
rewrite(output);
readln(n,m);
writeln(n,' ',m);
for i:=1 to n do
begin
for j:=1 to m do
begin
read(a);
write(a,' ');
end;
writeln;
end;
readln(n);
writeln(n);
for i:=1 to n do
begin
for j:=1 to m do
begin
read(a);
write(a,' ');
end;
writeln;
read(a);
writeln(a);
end;
close(output);
end;
end.

【对拍造数据】

#include<cstdio>
#include<cstdlib>
#include<ctime>
using namespace std;
int main()
{
freopen("T.in","w",stdout);
srand((int)time(0));
int n=50000,m=4,i,j;
printf("%d %d\n",n,m);
for (i=1;i<=n;i++)
{
for (j=1;j<=m;j++)
printf("%d ",rand()%10000+1);
printf("\n");
}
int Q=1000;
printf("%d\n",Q);
while (Q--)
{
for (i=1;i<=m;i++)
printf("%d ",rand()%10000+1);
printf("%d\n",rand()%5+1);
}
return 0;
}

【AC代码】

#include<cstdio>
#include<algorithm>
#include<queue>
#define N 50005
#define INF 21390627567143.0
using namespace std;
const int orzSYC=1;
struct arr
{
int d[5],max[15],min[15],l,r,id;
arr() {l=0;r=0;id=0;}
}a[N*4],aa[N];
struct pop
{
double x;int id;
friend bool operator < (const pop &a,const pop &b){return a.x<b.x;}
};
priority_queue<pop>ans;
int n,m,Q,i,j,t,x[15],D,temp[21],root,opt,P,flag;
inline int Read()
{
int x=0;char ch=getchar();bool positive=1;
for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') positive=0;
for (;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return positive?x:-x;
}
inline int cmp(arr a,arr b)
{
return a.d[D]<b.d[D];
}
inline void up(int k,int s)
{
for (int i=0;i<m;i++)
{
a[k].min[i]=min(a[k].min[i],a[s].min[i]);
a[k].max[i]=max(a[k].max[i],a[s].max[i]);
}
}
int build(int l,int r,int dd)
{
D=dd;int mid=((l+r)>>1);
nth_element(aa+l,aa+mid,aa+r+1,cmp);
for (int i=0;i<m;i++)
a[mid].min[i]=a[mid].max[i]=a[mid].d[i]=aa[mid].d[i];
a[mid].id=mid;
if (l<mid) a[mid].l=build(l,mid-1,(dd+1)%m);else a[mid].l=0;
if (mid<r) a[mid].r=build(mid+1,r,(dd+1)%m);else a[mid].r=0;
if (a[mid].l) up(mid,a[mid].l);
if (a[mid].r) up(mid,a[mid].r);
return mid;
}
inline double sdis(int k)
{
double res=0;
for (i=0;i<m;i++)
{
res+=(a[k].d[i]-x[i])*(a[k].d[i]-x[i]);
}
return res;
}
void ask(int k,int deep)
{
int L=a[k].l,R=a[k].r;
if (x[deep]>=a[k].d[deep]) swap(L,R);
double now=sdis(k);
if (L) ask(L,(deep+1)%m);
int flag=0;
if (ans.size()<P) {ans.push((pop){now,k});flag=1;}
else
{
if (now<ans.top().x) ans.pop(),ans.push((pop){now,k});
if ((x[deep]-a[k].d[deep])*1.*(x[deep]-a[k].d[deep])<ans.top().x) flag=1;
}
if (flag&&R) ask(R,(deep+1)%m);
}
int main()
{
while (scanf("%d%d",&n,&m)!=EOF)
{
for (i=1;i<=n;i++)
for (j=0;j<m;j++)
aa[i].d[j]=Read();
root=build(1,n,0);
Q=Read();
while (Q--)
{
for (i=0;i<m;i++) x[i]=Read();
P=Read();
ask(root,0);int wri=0;
printf("the closest %d points are:\n",P);
while (!ans.empty())
{
temp[++wri]=ans.top().id;
ans.pop();
}
for (i=wri;i;i--)
{
for (j=0;j<m-1;j++)
printf("%d ",a[temp[i]].d[j]);
printf("%d\n",a[temp[i]].d[m-1]);
}
}
}
return 0;

版权声明:本文博主原创文章。博客,未经同意不得转载。

kd-tree注解 &amp; bzoj 2648 &amp; 2716 &amp; 3053 解决问题的方法的更多相关文章

  1. BZOJ 2648: SJY摆棋子(K-D Tree)

    Time Limit: 20 Sec  Memory Limit: 128 MBSubmit: 6051  Solved: 2113[Submit][Status][Discuss] Descript ...

  2. 【BZOJ-2648&2716】SJY摆棋子&天使玩偶 KD Tree

    2648: SJY摆棋子 Time Limit: 20 Sec  Memory Limit: 128 MBSubmit: 2459  Solved: 834[Submit][Status][Discu ...

  3. BZOJ 3489: A simple rmq problem(K-D Tree)

    Time Limit: 40 Sec  Memory Limit: 512 MBSubmit: 2579  Solved: 888[Submit][Status][Discuss] Descripti ...

  4. BZOJ 3053: The Closest M Points(K-D Tree)

    Time Limit: 10 Sec  Memory Limit: 128 MBSubmit: 1235  Solved: 418[Submit][Status][Discuss] Descripti ...

  5. BZOJ 4520: [Cqoi2016]K远点对(k-d tree)

    Time Limit: 30 Sec  Memory Limit: 512 MBSubmit: 1162  Solved: 618[Submit][Status][Discuss] Descripti ...

  6. BZOJ 1941: [Sdoi2010]Hide and Seek(k-d Tree)

    Time Limit: 16 Sec  Memory Limit: 162 MBSubmit: 1712  Solved: 932[Submit][Status][Discuss] Descripti ...

  7. BZOJ2648/2716:SJY摆棋子/[Violet]天使玩偶(K-D Tree)

    Description 这天,SJY显得无聊.在家自己玩.在一个棋盘上,有N个黑色棋子.他每次要么放到棋盘上一个黑色棋子,要么放上一个白色棋子,如果是白色棋子,他会找出距离这个白色棋子最近的黑色棋子. ...

  8. K-D Tree题目泛做(CXJ第二轮)

    题目1: BZOJ 2716 题目大意:给出N个二维平面上的点,M个操作,分为插入一个新点和询问到一个点最近点的Manhatan距离是多少. 算法讨论: K-D Tree 裸题,有插入操作. #inc ...

  9. k-d tree模板练习

    1. [BZOJ]1941: [Sdoi2010]Hide and Seek 题目大意:给出n个二维平面上的点,一个点的权值是它到其他点的最长距离减最短距离,距离为曼哈顿距离,求最小权值.(n< ...

随机推荐

  1. Windows Phone开发(24):启动器与选择器之发送短信

    原文:Windows Phone开发(24):启动器与选择器之发送短信 本节我们通过一个简单的发送短信示例来演示一下如果配合使用PhoneNumberChooserTask和SmsComposeTas ...

  2. Linux 软连接与硬连接

    Linux 软连接与硬连接 对于一个文件来说,有唯一的索引接点与之相应,而对于一个索引接点号,却能够有多个文件名称与之相应.因此,在磁盘上的同一个文件能够通过不同的路径去訪问该文件.注意在Linux下 ...

  3. Quick StateMachine状态机

    状态机quick中是一个亮点,假设我们做一款RPG游戏,一个角色通常会拥有idle,attack,walk.run,death这些状态,假设游戏角色的状态採用分支条件推断的话.会造成很庞大而难以维护. ...

  4. Codeforces 432 D. Prefixes and Suffixes

    用扩展KMP做简单省力..... D. Prefixes and Suffixes time limit per test 1 second memory limit per test 256 meg ...

  5. BackTrack5 (BT5)无线password破解教程WPA/WPA2-PSK无线password皴

    昨天公布了BackTrack5 (BT5)无线weppassword破解教程之minidwep-gtk破解法一文,对BT5下破解wep无线password的简单方法做了介绍,今天奶牛为朋友们介绍下怎样 ...

  6. silverlight与wcf双向通讯 例子

    本文将建立一个silverlight与wcf双向通讯的简单实例,以下是详细步骤: 新建Silverlight应用程序,名称WCFtest.解决方案中添加WCF服务应用程序,名称WcfServiceTe ...

  7. twrp 2.7.0 ui.xml简单分析,布局讲解,第一章

    twrp 的ui.xml文件在bootable/recovery/gui/devices/$(DEVICE_RESOLUTION)/res目录里面 下面我主要分析的是720x1280分辨率的界面布局及 ...

  8. C++ 版本的split_string

    vector<string> split_string(const string &in, char del, bool skip_empty) { vector<strin ...

  9. 数据结构c字符串操作语言版本

    #include<stdio.h> #include<malloc.h> #include<string.h> //结构的定义字符串 typedef struct ...

  10. 我理解设计模式C++实现观察者模式Observer Pattern

    概述: 近期中国股市起起伏伏,当然了起伏就用商机,小明发现商机后果断想入市,买入了中国证券,他想在电脑client上,网页上,手机上,iPad上都能够查看到该证券的实时行情,这样的情况下我们应该怎么设 ...