BZOJ 2648: SJY摆棋子 (动态kd-tree,插入,曼哈顿距离,输入挂)
Description
这天,SJY显得无聊。在家自己玩。在一个棋盘上,有N个黑色棋子。他每次要么放到棋盘上一个黑色棋子,要么放上一个白色棋子,如果是白色棋子,他会找出距离这个白色棋子最近的黑色棋子。此处的距离是 曼哈顿距离 即(|x1-x2|+|y1-y2|) 。现在给出N<=500000个初始棋子。和M<=500000个操作。对于每个白色棋子,输出距离这个白色棋子最近的黑色棋子的距离。同一个格子可能有多个棋子。
Input
第一行两个数 N M
以后M行,每行3个数 t x y
如果t=1 那么放下一个黑色棋子
如果t=2 那么放下一个白色棋子
Output
对于每个T=2 输出一个最小距离
Sample Input
2 3 1 1 2 3 2 1 2 1 3 3 2 4 2
Sample Output
1 2
其实就是BZOJ2716 的双倍经验题
写出来是因为,这道题要加输入挂才可以过orz
/* ***********************************************
Author :111qqz
Created Time :2017年10月10日 星期二 13时35分26秒
File Name :2716.cpp
************************************************ */
1#include <cstdio>
2#include <cstring>
3#include <iostream>
4#include <algorithm>
5#include <vector>
6#include <queue>
7#include <set>
8#include <map>
9#include <string>
10#include <cmath>
11#include <cstdlib>
12#include <ctime>
13#define PB push_back
14#define fst first
15#define sec second
16#define lson l,m,rt<<1
17#define rson m+1,r,rt<<1|1
18#define ms(a,x) memset(a,x,sizeof(a))
19typedef long long LL;
20#define pi pair < int ,int >
21#define MP make_pair
1using namespace std;
2const double eps = 1E-8;
3const int dx4[4]={1,0,0,-1};
4const int dy4[4]={0,-1,1,0};
5const int inf = 0x3f3f3f3f;
6const LL linf = 1LL<<60;
7const int N=5E5+7;
8int n,m;
9int idx,rt;
10LL ans;
11LL getLL() {
12 LL k = 0, fh = 1; char c = getchar();
13 for(; c < '0' || c > '9'; c = getchar())
14 if (c == '-') fh = -1;
15 for(; c >= '0' && c <= '9'; c = getchar())
16 k = k * 10 + c - '0';
17 return k * fh;
18}
19struct KDT
20{
21 LL coor[2];
22 LL mn[2],mx[2]; //需要维护四个方向的最值,是因为是曼哈顿距离。
23 int son[2];
24 bool operator < (const KDT &u)const{ return coor[idx]<u.coor[idx];}
25 void init() { for ( int i = 0 ; i < 2 ;i++) mn[i]=mx[i]=coor[i];}
26 void input() { for ( int i = 0 ; i < 2 ; i++) coor[i]=getLL();}
27}v[N<<1],tar; //开一倍空间是因为可能全都是插点。
28inline LL getDis(KDT a,KDT b) {return abs(a.coor[0]-b.coor[0]) + abs(a.coor[1]-b.coor[1]);}
29void up( int x)
30{
31 for ( int i = 0 ; i < 2 ; i++) if (v[x].son[i]){
32 int y =v[x].son[i];
33 for (int j = 0 ; j < 2 ; j++)
34 v[x].mn[j]=min(v[x].mn[j],v[y].mn[j]),v[x].mx[j]=max(v[x].mx[j],v[y].mx[j]);
35 }
36}
37int build( int l=1,int r=n,int dim=0)
38{
39 idx = dim;
40 int mid = (l+r)>>1;
41 nth_element(v+l,v+mid,v+r+1);
42 v[mid].init();
43 if (l!=mid) v[mid].son[0] = build(l,mid-1,1-dim);
44 if (r!=mid) v[mid].son[1] = build(mid+1,r,1-dim);
45 up(mid);
46 return mid;
47}
48LL ask( int x,KDT p)
49{
50 LL ret=0;
51 for ( int i = 0 ; i < 2 ; i++)
52 {
53 ret = ret +max(0LL,v[x].mn[i]-p.coor[i])+ max(0LL,p.coor[i]-v[x].mx[i]);
54 }
55 return ret;
56}
57void insert(int x=rt,int dim=0)
58{
59 bool fg=v[n].coor[dim]>v[x].coor[dim];
60 if (v[x].son[fg]) insert(v[x].son[fg],1-dim);
61 else v[x].son[fg]=n;
62 up(x);
63}
64void query(int x=rt,int dim=0)
65{
66 LL dis=getDis(v[x],tar),dl=linf,dr=linf;
67 ans=min(dis,ans);
68 if (v[x].son[0]) dl=ask(v[x].son[0],tar);
69 if (v[x].son[1]) dr=ask(v[x].son[1],tar);
70 int xx = v[x].son[0],yy=v[x].son[1];
71 if (dl>=dr) swap(dl,dr),swap(xx,yy);
72 if (dl<ans) query(xx,1-dim);
73 if (dr<ans) query(yy,1-dim);
74}
1int main()
2{
3 #ifndef ONLINE_JUDGE
4 freopen("./in.txt","r",stdin);
5 #endif
6 cin>>n>>m;
7 for ( int i = 1 ; i <= n ; i++) v[i].input();
8 rt =build();
9 while (m--)
10 {
11 int t;
12 scanf("%d",&t);
13 ans=linf;
14 if (t==1) v[++n].input(),v[n].init(),insert();
15 else tar.input(),query(),printf("%lld\n",ans);
16 }
17 #ifndef ONLINE_JUDGE
18 fclose(stdin);
19 #endif
20 return 0;
21}