[bzoj4154][Ipsc2015]Generating Synergy_KD-Tree_dfs序

Generating Synergy bzoj-4154 Ipsc-2015

题目大意：给定一棵n个节点树，m个操作，支持：将一个点周围所有距该点距离不超过l的子结点的颜色改成另一种颜色；查询单点颜色。每个点的颜色开始都是1。

注释：$1\le n,m\le 10^5$

想法：这就是一个典型的不是KD-Tree裸题，我们需要将它转化成点。

首先，我们先拉出整棵树的dfs序。

每个节点，我们设横坐标就是dfs序上的下标，纵坐标就是深度。我们发现，一个点的子树是一段连续的区间，然后当深度确定是，我们发现这就是矩阵赋值，加上pushdown操作即可。

单点查询遍历即可。不要忘记query的时候也要pushdown

另外没有插入操作，就不用重构了。

最后，附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 100010
using namespace std;
int head
,to
,nxt
,cnt,deep
,pos
,last
,tot,d,root;
struct data
{
int p[2],mx[2],mn[2],c[2],w,tag;
bool operator <(const data &a) const
{
return p[d]==a.p[d]?p[d^1]<a.p[d^1]:p[d]<a.p[d];
}
}a
;
inline void add(int x,int y)
{
to[++cnt]=y,nxt[cnt]=head[x],head[x]=cnt;
}
void dfs(int x)
{
int i;
pos[x]=++tot,a[x].p[0]=pos[x],a[x].p[1]=deep[x];
for(i=head[x];i;i=nxt[i])
deep[to[i]]=deep[x]+1,dfs(to[i]);
last[x]=tot;
}
inline void pushup(int x)
{
int l=a[x].c[0],r=a[x].c[1];
a[x].mx[0]=max(a[x].p[0],max(a[l].mx[0],a[r].mx[0]));
a[x].mx[1]=max(a[x].p[1],max(a[l].mx[1],a[r].mx[1]));
a[x].mn[0]=min(a[x].p[0],min(a[l].mn[0],a[r].mn[0]));
a[x].mn[1]=min(a[x].p[1],min(a[l].mn[1],a[r].mn[1]));
}
int build(int l,int r,int now)
{
if(l>r) return 0;
int mid=(l+r)>>1;
d=now,nth_element(a+l,a+mid,a+r+1);
a[mid].w=1,a[mid].tag=0;
a[mid].c[0]=build(l,mid-1,now^1);
a[mid].c[1]=build(mid+1,r,now^1);
pushup(mid);
return mid;
}
inline void pushdown(int x)
{
if(a[x].tag)
{
int l=a[x].c[0],r=a[x].c[1];
a[l].w=a[l].tag=a[r].w=a[r].tag=a[x].tag;
a[x].tag=0;
}
}
void update(int bx,int ex,int by,int ey,int v,int x)
{
if(!x||a[x].mx[0]<bx||a[x].mn[0]>ex||a[x].mx[1]<by||a[x].mn[1]>ey)return;
if(a[x].mn[0]>=bx&&a[x].mx[0]<=ex&&a[x].mn[1]>=by&&a[x].mx[1]<=ey)
{
a[x].w=a[x].tag=v;
return;
}
pushdown(x);
if(a[x].p[0]>=bx&&a[x].p[0]<=ex&&a[x].p[1]>=by&&a[x].p[1]<=ey)a[x].w=v;
update(bx,ex,by,ey,v,a[x].c[0]),update(bx,ex,by,ey,v,a[x].c[1]);
}
int query(int px,int py,int x)
{
d^=1;
if(a[x].p[0]==px&&a[x].p[1]==py) return a[x].w;
pushdown(x);
if(d)
{
if(py<a[x].p[1]||(py==a[x].p[1]&&px<a[x].p[0])) return query(px,py,a[x].c[0]);
else return query(px,py,a[x].c[1]);
}
else
{
if(px<a[x].p[0]||(px==a[x].p[0]&&py<a[x].p[1])) return query(px,py,a[x].c[0]);
else return query(px,py,a[x].c[1]);
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(head,0,sizeof(head)),cnt=1;
a[0].mx[0]=a[0].mx[1]=-1<<30,a[0].mn[0]=a[0].mn[1]=1<<30;
int n,m,i,x,y,z,ans=0;
scanf("%d%*d%d",&n,&m);
for(i=2;i<=n;i++) scanf("%d",&x),add(x,i);
dfs(1);
root=build(1,n,0);
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&z);
if(z) update(pos[x],last[x],deep[x],deep[x]+y,z,root);
else d=1,ans=(ans+(long long)query(pos[x],deep[x],root)*i)%1000000007;
}
printf("%d\n",ans);
}
return 0;
}

小结：KD-Tree这种题还是挺好玩的..