【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\n",ans);
		}
	}
	return 0;
}