bzoj 4066: 简单题 （KD-tree）

4066: 简单题

Time Limit: 50 Sec  Memory Limit: 20 MB
Submit: 2225  Solved: 592
[ Submit][ Status][ Discuss]

Description

你有一个N*N的棋盘，每个格子内有一个整数，初始时的时候全部为0，现在需要维护两种操作：

 

命令	参数限制	内容
1 x y A	1<=x,y<=N，A是正整数	将格子x,y里的数字加上A
2 x1 y1 x2 y2	1<=x1<= x2<=N 1<=y1<= y2<=N	输出x1 y1 x2 y2这个矩形内的数字和
3	无	终止程序

Input

输入文件第一行一个正整数N。

接下来每行一个操作。每条命令除第一个数字之外，

均要异或上一次输出的答案last_ans，初始时last_ans=0。

Output

对于每个2操作，输出一个对应的答案。

Sample Input

4
1 2 3 3
2 1 1 3 3
1 1 1 1
2 1 1 0 7
3

Sample Output

3
5

HINT

数据规模和约定

1<=N<=500000,操作数不超过200000个，内存限制20M，保证答案在int范围内并且解码之后数据仍合法。

样例解释见OJ2683

新加数据一组，但未重测----2015.05.24

Source

By wjy1998

[ Submit][ Status][ Discuss]

题解：KD-tree

这道题据说写的太丑会被卡。。。刚开始TLE了三次。

因为我们是动态的加入点，所以很有可能会破坏KD-tree的平衡结构。而KD-tree时间复杂度的保证就是平衡的树结构，所以我们经过一段时间就要重建一次。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 200003
using namespace std;
int n,m,root,cmpd,x,y,x0,y0,a,ans;
struct data{
	int d[2],mx[2],mn[2],sum,l,r,val;
}tr[N];
int cmp(data a,data b)
{
	return a.d[cmpd]<b.d[cmpd]||a.d[cmpd]==b.d[cmpd]&&a.d[cmpd^1]<b.d[cmpd^1];
}
void update(int now)
{
	int l=tr[now].l; int r=tr[now].r;
	for (int i=0;i<=1;i++) {
		if (l) tr[now].mx[i]=max(tr[now].mx[i],tr[l].mx[i]),
		       tr[now].mn[i]=min(tr[now].mn[i],tr[l].mn[i]);
		if (r) tr[now].mx[i]=max(tr[now].mx[i],tr[r].mx[i]),
		       tr[now].mn[i]=min(tr[now].mn[i],tr[r].mn[i]);
	}
	if (l) tr[now].sum+=tr[l].sum;
	if (r) tr[now].sum+=tr[r].sum;
}
int rebuild(int l,int r,int d)
{
	cmpd=d;
	int mid=(l+r)/2;
	nth_element(tr+l,tr+mid,tr+r+1,cmp);
	for (int i=0;i<=1;i++)
	 tr[mid].mx[i]=tr[mid].mn[i]=tr[mid].d[i];
	tr[mid].sum=tr[mid].val;
	tr[mid].l=tr[mid].r=0;
	if (l<mid) tr[mid].l=rebuild(l,mid-1,d^1);
	if (r>mid) tr[mid].r=rebuild(mid+1,r,d^1);
	update(mid);
	return mid;
}
void insert(int now)
{
	if (!root) {
		root=now;
		return;
	}
	int x=root; int d=0;
	while (true) {
		for (int i=0;i<=1;i++)
		 tr[x].mx[i]=max(tr[x].mx[i],tr[now].mx[i]),
		 tr[x].mn[i]=min(tr[x].mn[i],tr[now].mn[i]);
		tr[x].sum+=tr[now].sum;
		//cout<<tr[x].d[0]<<" "<<tr[x].d[1]<<" "<<tr[x].sum<<endl;
		if (tr[now].d[d]>=tr[x].d[d]) {
			if (!tr[x].r) {
				tr[x].r=now;
				return;
			}else x=tr[x].r;
		}
		else {
			if (!tr[x].l) {
				tr[x].l=now;
				return;
			}
			else x=tr[x].l;
		}
		d^=1;
	}
}
int pd(int now)
{
	int nowx=tr[now].d[0]; int nowy=tr[now].d[1];
	if (nowx>=x&&nowx<=x0&&nowy>=y&&nowy<=y0) return 1;
	else return 0;
}
int check(int now)
{
	if (tr[now].mx[0]<=x0&&tr[now].mn[0]>=x&&tr[now].mx[1]<=y0&&tr[now].mn[1]>=y) return 1;
	if (tr[now].mx[0]<x||tr[now].mn[0]>x0||tr[now].mx[1]<y||tr[now].mn[1]>y0) return -1;
	return 0;
}
void query(int now)
{
	if (pd(now)) ans+=tr[now].val;
	if (tr[now].l) {
		int t=check(tr[now].l);
		if (t==1) ans+=tr[tr[now].l].sum;
		else if (t==0) query(tr[now].l); 
	}
	if (tr[now].r) {
		int t=check(tr[now].r);
		if (t==1) ans+=tr[tr[now].r].sum;
		else if (t==0) query(tr[now].r);
	}
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d",&m);
	while (true) {
		int opt; scanf("%d",&opt);
		if (opt==3) break;
		if (opt==1) {
			scanf("%d%d%d",&x,&y,&a); x^=ans; y^=ans; a^=ans;
			++n;
			tr[n].d[0]=tr[n].mx[0]=tr[n].mn[0]=x; 
			tr[n].d[1]=tr[n].mx[1]=tr[n].mn[1]=y; 
			tr[n].val=tr[n].sum=a;
			insert(n);
			if (n%10000==0) 
			  root=rebuild(1,n,0);
		}
		if (opt==2) {
			scanf("%d%d%d%d",&x,&y,&x0,&y0);
			x0^=ans; y0^=ans; x^=ans; y^=ans;
			//cout<<x0<<" "<<y0<<" "<<x<<" "<<y<<endl;
			ans=0;
			query(root);
			printf("%d\n",ans);
		}
	}
}