【BZOJ2989】数列 kd-tree

【BZOJ2989】数列

Description

给定一个长度为n的正整数数列a[i]。

定义2个位置的graze值为两者位置差与数值差的和，即graze(x,y)=|x-y|+|a[x]-a[y]|。

2种操作（k都是正整数）：

1.Modify x k：将第x个数的值修改为k。

2.Query x k：询问有几个i满足graze(x,i)<=k。因为可持久化数据结构的流行，询问不仅要考虑当前数列，还要

考虑任意历史版本，即统计任意位置上出现过的任意数值与当前的a[x]的graze值<=k的对数。（某位置多次修改为

同样的数值，按多次统计）

Input

第1行两个整数n,q。分别表示数列长度和操作数。

第2行n个正整数，代表初始数列。

第3--q+2行每行一个操作。

Output

对于每次询问操作，输出一个非负整数表示答案

Sample Input

3 5
2 4 3
Query 2 2
Modify 1 3
Query 2 2
Modify 1 2
Query 1 1

Sample Output

2
3
3

HINT

N<=60000 修改操作数<=40000 询问<=50000 Max{a[i]}含修改<=100000

题解：新技能：旋转坐标系。就是将一个斜着的正方形变成横着的，只需要将所有点（x,y）变成（x+y，y-x）即可。

然后就变成了查询所有满足一下条件的点：l<=x<=r，l<=y<=r，加入时间<=询问时间。直接上三维kd-tree就好了。

本机实测一个点8s+然而提交上去5s就A了是什么鬼~

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define D1 ((D+1)%3)
#define D2 ((D+2)%3)
#define rep for(int i=0;i<=2;i++)
using namespace std;
const int maxn=100010;
struct kd
{
	int v[3],sm[3],sn[3],siz,ls,rs;
	kd (){}
	int & operator [] (int x)	{return v[x];}
	kd (int a,int b,int c)	{v[0]=sm[0]=sn[0]=a,v[1]=sm[1]=sn[1]=b,v[2]=sm[2]=sn[2]=c,siz=1,ls=rs=0;}
};
kd t[maxn];
int n,m,D,tot,ans,root,cnt;
int qx[maxn],qk[maxn],X[maxn],Y[maxn],Z[maxn],head[maxn],rm[3],rn[3];
bool cmp(kd a,kd b)
{
	return (a[D]==b[D])?((a[D1]==b[D1])?(a[D2]<b[D2]):(a[D1]<b[D1])):(a[D]<b[D]);
}
int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
	return ret*f;
}
void pushup(int x,int y)
{
	t[x].siz+=t[y].siz;
	rep t[x].sm[i]=max(t[x].sm[i],t[y].sm[i]),t[x].sn[i]=min(t[x].sn[i],t[y].sn[i]);
}
int build(int l,int r,int d)
{
	if(l>r)	return 0;
	D=d;
	int mid=l+r>>1;
	nth_element(t+l,t+mid,t+r+1,cmp);
	t[mid].ls=build(l,mid-1,(d+1)%3),t[mid].rs=build(mid+1,r,(d+1)%3);
	if(t[mid].ls)	pushup(mid,t[mid].ls);
	if(t[mid].rs)	pushup(mid,t[mid].rs);
	return mid;
}
int getmin(int x)
{
	int ret=0;
	rep	ret|=(t[x].sm[i]<rn[i]||t[x].sn[i]>rm[i]);
	return !ret;
}
int getmax(int x)
{
	int ret=1;
	rep ret&=(t[x].sm[i]<=rm[i]&&t[x].sn[i]>=rn[i]);
	return ret;
}
int check(int x)
{
	int ret=1;
	rep ret&=(t[x].v[i]<=rm[i]&&t[x].v[i]>=rn[i]);
	return ret;
}
void query(int x)
{
	if(!x||!getmin(x))	return ;
	if(getmax(x))
	{
		ans+=t[x].siz;
		return ;
	}
	if(check(x))	ans++;
	query(t[x].ls),query(t[x].rs);
}
char str[10];
int main()
{
	n=rd(),m=rd();
	int i,a,b;
	for(i=1;i<=n;i++)	a=i,b=rd(),X[i]=a+b,Y[i]=b-a,Z[i]=0,head[i]=i;
	for(tot=n,i=1;i<=m;i++)
	{
		scanf("%s",str);
		if(str[0]=='M')	a=rd(),b=rd(),X[++tot]=a+b,Y[tot]=b-a,Z[tot]=i,head[a]=tot;
		else	qx[i]=head[rd()],qk[i]=rd();
	}
	for(i=1;i<=tot;i++)	t[i]=kd(X[i],Y[i],Z[i]);
	root=build(1,tot,0);
	for(i=1;i<=m;i++)
	{
		if(qx[i])
		{
			rm[0]=X[qx[i]]+qk[i],rn[0]=X[qx[i]]-qk[i];
			rm[1]=Y[qx[i]]+qk[i],rn[1]=Y[qx[i]]-qk[i];
			rm[2]=i,rn[2]=0;
			ans=0,query(root),printf("%d
",ans);
		}
	}
	return 0;
}