BZOJ 4154: [Ipsc2015]Generating Synergy KDtree+dfs序

多组数据真tm恶心~

把 $dfs$序和深度分别看作横纵坐标，然后用 $KDtree$ 数点就可以了~

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 300005
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int d;
struct Node {
	int ch[2], tag, col, minv[2], maxv[2], p[2];
}t[N];
int isout(int x1,int y1,int x2,int y2,int x) {
	return (x1>t[x].maxv[0]||x2<t[x].minv[0]||y1>t[x].maxv[1]||y2<t[x].minv[1]);
}
int isin(int x1,int y1,int x2,int y2,int x) {
	return (t[x].minv[0]>=x1&&t[x].maxv[0]<=x2&&t[x].minv[1]>=y1&&t[x].maxv[1]<=y2);
}
bool cmp(Node a,Node b) {
	return a.p[d]==b.p[d]?a.p[d^1]<b.p[d^1]:a.p[d]<b.p[d];
}
void pushup(int x,int y) {
	for(int i=0;i<2;++i)
		t[x].minv[i]=min(t[x].minv[i],t[y].minv[i]),t[x].maxv[i]=max(t[x].maxv[i],t[y].maxv[i]);
}
void mark(int x,int c) {
	t[x].col=t[x].tag=c;
}
void pushdown(int x) {
	if(t[x].tag!=-1) {
	    if(t[x].ch[0]) mark(t[x].ch[0],t[x].tag);
	    if(t[x].ch[1]) mark(t[x].ch[1],t[x].tag);
	    t[x].tag=-1;
	}
}
int build(int l,int r,int o) {
	int mid=(l+r)>>1,i;
	d=o, nth_element(t+l,t+mid,t+1+r,cmp);
	t[mid].minv[0]=t[mid].maxv[0]=t[mid].p[0];
	t[mid].minv[1]=t[mid].maxv[1]=t[mid].p[1];
	t[mid].col=1, t[mid].tag=-1;
	if(mid>l) t[mid].ch[0]=build(l,mid-1,o^1), pushup(mid, t[mid].ch[0]);
	if(r>mid)  t[mid].ch[1]=build(mid+1,r,o^1), pushup(mid, t[mid].ch[1]);
	return mid;
}
void update(int x1,int y1,int x2,int y2,int c,int x) {
	if(!x||isout(x1,y1,x2,y2,x)) return;
	if(isin(x1,y1,x2,y2,x)) {
		mark(x,c);
		return;
	}
	if(t[x].p[0]>=x1&&t[x].p[0]<=x2&&t[x].p[1]>=y1&&t[x].p[1]<=y2) t[x].col=c;
	pushdown(x);
	if(t[x].ch[0])
		update(x1,y1,x2,y2,c,t[x].ch[0]);
	if(t[x].ch[1])
		update(x1,y1,x2,y2,c,t[x].ch[1]);
}
int query(int x1,int y1,int x) {
	if(!x||isout(x1,y1,x1,y1,x)) return -1;
	if(t[x].p[0]==x1&&t[x].p[1]==y1) return t[x].col;
	pushdown(x);
	return max(query(x1,y1,t[x].ch[0]), query(x1,y1,t[x].ch[1]));
}
int n,edges,tim,fa[N],hd[N],nex[N],to[N],dfn[N],size[N],dep[N];
void addedge(int u,int v) {
	nex[++edges]=hd[u],hd[u]=edges,to[edges]=v;
}
void dfs(int u) {
	dfn[u]=++tim,size[u]=1;
	for(int i=hd[u];i;i=nex[i])
		dep[to[i]]=dep[u]+1,dfs(to[i]),size[u]+=size[to[i]];
}
void solve() {
	edges=0;
	memset(hd,0,sizeof(hd));
	int i,j;
	int Q,C;
	scanf("%d%d%d",&n,&C,&Q);
	for(i=2;i<=n;++i) scanf("%d",&fa[i]), addedge(fa[i],i);
	dfs(1);
    for(i=1;i<=n;++i) t[i].p[0]=dfn[i],t[i].p[1]=dep[i],t[i].ch[0]=t[i].ch[1]=0, t[i].tag=-1;
    int root=build(1,n,0);
    long long lastans=0;
    for(i=1;i<=Q;++i) {
    	int a,l,c;
    	scanf("%d%d%d",&a,&l,&c);
    	if(c==0) lastans=(lastans+(long long)i*query(dfn[a], dep[a], root))%1000000007;
    	else {
    		update(dfn[a],dep[a],dfn[a]+size[a]-1,dep[a]+l,c,root);
    	}
    }
    printf("%lld\n",lastans);
}
int main() {
	int T;
	// setIO("input");
	scanf("%d",&T);
	while(T--) solve();
	return 0;
}