zoukankan      html  css  js  c++  java
  • bzoj 2648: SJY摆棋子【KD-tree】

    其实理论上cdq更优
    核心是依次取x值、y值的mid作为当前节点,向两边递归建立二叉树,树上维护size:子树大小;mx[0/1]:子树内最大x/y;mn[0/1]:子树内最小x/y;d[0/1]:这个点的x/y;
    建树的时候用到nth_element,用处是把第k个数放到k位置,并且把小于k的放在k前,大于k的放在k后(算是快排的一部分)
    插入点的时候,像走二叉搜索树一样往下走(不过每走一步key值就要变,xy交替),然后走到空就把新点挂上去
    查找的时候比较像退火,通过mnmx找到子树能更新答案的最小值预估,然后决定要不要往下走,并且一路走一路用真实值更新答案;如果左右儿子都能走,就先走估值小的那边
    然后出现了一个问题,这个插入有可能把二叉树变成一条链,这个时候要用替罪羊树的思想,设一个阈值,如果左右儿子的size比例超过阈值就把这颗子树拍扁重构

    #include<iostream>
    #include<cstdio>
    #include<algorithm>
    using namespace std;
    const int N=500005;
    const double alp=0.75;
    int n,m,w,tot,rt,ans;
    struct qwe
    {
    	int a[2];
    	int& operator [] (int x)
    	{
    		return a[x];
    	}
    	bool operator < (const qwe &b) const
    	{
    		return a[w]<b.a[w];
    	}
    }a[N];
    struct KD
    {
    	int s,ls,rs,mn[2],mx[2];
    	qwe d;
    }t[N*5];
    int read()
    {
    	int r=0,f=1;
    	char p=getchar();
    	while(p>'9'||p<'0')
    	{
    		if(p=='-')
    			f=-1;
    		p=getchar();
    	}
    	while(p>='0'&&p<='9')
    	{
    		r=r*10+p-48;
    		p=getchar();
    	}
    	return r*f;
    }
    void minn(int &x,int y)
    {
    	x>y?x=y:0;
    }
    void maxx(int &x,int y)
    {
    	x<y?x=y:0;
    }
    void ud(int ro)
    {
    	if(t[ro].ls)
    		for(int i=0;i<=1;i++)
    			minn(t[ro].mn[i],t[t[ro].ls].mn[i]),maxx(t[ro].mx[i],t[t[ro].ls].mx[i]);
    	if(t[ro].rs)
    		for(int i=0;i<=1;i++)
    			minn(t[ro].mn[i],t[t[ro].rs].mn[i]),maxx(t[ro].mx[i],t[t[ro].rs].mx[i]);
    	t[ro].s=t[t[ro].ls].s+t[t[ro].rs].s+1;
    }
    int build(int l,int r,int f)
    {
    	if(l>r)
    		return 0;
    	w=f;
    	int mid=(l+r)>>1,ro=++tot;
    	nth_element(a+l,a+mid,a+r+1);
    	t[ro].s=1;
    	t[ro].d[0]=t[ro].mn[0]=t[ro].mx[0]=a[mid][0];
    	t[ro].d[1]=t[ro].mn[1]=t[ro].mx[1]=a[mid][1];
    	t[ro].ls=build(l,mid-1,f^1);
    	t[ro].rs=build(mid+1,r,f^1);
    	ud(ro);
    	return ro;
    }
    void pia(int &ro,int s)
    {
    	if(t[ro].ls)
    		pia(t[ro].ls,s);
    	a[s+t[t[ro].ls].s+1]=t[ro].d;
    	if(t[ro].rs)
    		pia(t[ro].rs,s+t[t[ro].ls].s+1);
    }
    void ok(int &ro,int f)
    {
    	if(alp*t[ro].s<max(t[t[ro].ls].s,t[t[ro].rs].s))
    		pia(ro,0),ro=build(1,t[ro].s,f);
    }
    void update(int &ro,int x,int y,int f)
    {
    	if(!ro)
    	{
    		ro=++tot;
    		t[ro].s=1;
    		t[ro].mn[0]=t[ro].mx[0]=t[ro].d[0]=x;
    		t[ro].mn[1]=t[ro].mx[1]=t[ro].d[1]=y;
    		return;
    	}
    	if(t[ro].d[f]>=(f==0?x:y))
    		update(t[ro].ls,x,y,f^1);
    	else
    		update(t[ro].rs,x,y,f^1);
    	ud(ro);
    	ok(ro,f);
    }
    int dis(int ro,int x,int y)
    {
    	return max(x-t[ro].mx[0],0)+max(t[ro].mn[0]-x,0)+max(y-t[ro].mx[1],0)+max(t[ro].mn[1]-y,0);
    }
    void ques(int ro,int x,int y)
    {
    	ans=min(ans,abs(x-t[ro].d[0])+abs(y-t[ro].d[1]));
    	int dl=t[ro].ls?dis(t[ro].ls,x,y):1e9,dr=t[ro].rs?dis(t[ro].rs,x,y):1e9;
    	if(dl<dr)
    	{
    		if(dl<ans)
    			ques(t[ro].ls,x,y);
    		if(dr<ans)
    			ques(t[ro].rs,x,y);
    	}
    	else
    	{
    		if(dr<ans)
    			ques(t[ro].rs,x,y);
    		if(dl<ans)
    			ques(t[ro].ls,x,y);
    	}
    }
    int main()
    {
    	n=read(),m=read();
    	for(int i=1;i<=n;i++)
    		a[i][0]=read(),a[i][1]=read();
    	rt=build(1,n,0);
    	while(m--)
    	{
    		int o=read(),x=read(),y=read();
    		if(o==1)
    			update(rt,x,y,0);
    		else
    		{
    			ans=1e9;
    			ques(rt,x,y);
    			printf("%d
    ",ans);
    		}
    	}
    	return 0;
    }
    
  • 相关阅读:
    归并、希尔排序
    堆排序
    [模板] 最小树形图/朱刘算法
    [模板] 常系数线性递推
    [模板] Kruskal算法 && 克鲁斯卡尔重构树
    [模板] 斯特林数,性质以及求法
    这几天想干什么
    奇怪的 Markdown / LaTeX 笔记
    [模板] 各种莫队
    [模板] 2-SAT 问题
  • 原文地址:https://www.cnblogs.com/lokiii/p/10098394.html
Copyright © 2011-2022 走看看