一开始没思路 感觉像是一个树形dp 然而不会
然后看了一眼题解就明白了
一个点的子树 用dfs序表示肯定是一个连续的区间 并且由于有子树的距离限制 可以转化为一个深度的区间
于是每个点都会有一个在二维平面上的标号(x,y) == (编号,深度)
并且每次进行更新一个节点的子树的时候就可以得到一个x的区间和一个y的区间 (实际上这些x是独一无二的 y更像是一个限制的条件)
这样就可以转化成一个二维的平面 每次选择一个矩阵进行颜色的统一更新 询问单点的颜色 做一个延时标记
询问单点的时候可以每次找啊找直到找到这个点 也可以判断一下mark mark一定是最新的历史 如果这个点在这个区间里面并且这个区间有修改历史 那么ans就是mark
不过加了这个优化速度并没有提升多少
在这里kd-tree很像线段树 区别就是维度不同
找了一个小bug就直接a掉了 开心!
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<math.h>
#include<queue>
#include<string>
using namespace std;
#define L long long
const L INF = 999999999 ;
const L maxn = 100050 ;
const L mod = 1000000007;
/**
用dfs序设定每个点的子树的标号区间
存下每个点的deep
则每个点在二维平面的x为它的标号 y为它的深度 (如果仅要表示这个点主码x就可以了)
利用kdtree来进行区间修改单点查询
如果区间满足 Max[0] Min[0] 属于 x1 x2 && Max[1] Min[1] 属于 y11 y2 就直接修改这个区间的颜色 设置一个延时标记
x1 x2 == l[a] r[a] y11 y2 == deep[a] deep[a] + l
单点查询直接进行一个logn的查询
*/
L n , c , q ;
L deep[maxn] ;
vector<L >vec[maxn] ;
L l[maxn] , r[maxn] ;
L cnt ;
void dfs(L u , L dep) {
deep[u] = dep ;
l[u] = ++ cnt ;
for(L i = 0; i < vec[u].size(); i ++ ){
L v = vec[u][i] ;
dfs(v,dep + 1) ;
}
r[u] = cnt ;
}
L root , cmp_d ;
struct node {
L l , r ;
L d[2] , Max[2] , Min[2] ;
L c ;
L mark ;
}a[maxn];
bool cmp(node a,node b){return ((a.d[cmp_d] < b.d[cmp_d])||(a.d[cmp_d] == b.d[cmp_d] && a.d[!cmp_d] < b.d[!cmp_d])) ;} ;
void up(L p , L k ){
if(a[k].Max[0] > a[p].Max[0]) a[p].Max[0] = a[k].Max[0] ;
if(a[k].Max[1] > a[p].Max[1]) a[p].Max[1] = a[k].Max[1] ;
if(a[k].Min[0] < a[p].Min[0]) a[p].Min[0] = a[k].Min[0] ;
if(a[k].Min[1] < a[p].Min[1]) a[p].Min[1] = a[k].Min[1] ;
}
void down(L p) {
if(a[p].mark == 0)return ;
L newc = a[p].mark ;
a[p].mark = 0 ;
if(a[p].r) {
L rr = a[p].r ;
a[rr].c = newc ;
a[rr].mark = newc ;
}
if(a[p].l) {
L ll = a[p].l ;
a[ll].c = newc ;
a[ll].mark = newc ;
}
return ;
}
L build(L l, L r , L D) {
L mid = (l+r)/2 ;
cmp_d = D;
nth_element(a+1+l,a+1+mid,a+1+r,cmp) ;
a[mid].Max[0] = a[mid].Min[0] = a[mid].d[0] ;
a[mid].Max[1] = a[mid].Min[1] = a[mid].d[1] ;
a[mid].c = 1 ;
a[mid].mark = 0 ;
if(l != mid)a[mid].l = build(l,mid-1,D^1); else a[mid].l = 0;
if(r != mid)a[mid].r = build(mid+1,r,D^1); else a[mid].r = 0;
if(a[mid].l)up(mid,a[mid].l) ;
if(a[mid].r)up(mid,a[mid].r) ;
return mid ;
}
void upda(L p , L x1 , L y11 , L x2 , L y2 , L newc ) {
if(a[p].Max[0] < x1 || a[p].Min[0] > x2 || a[p].Max[1] < y11 || a[p].Min[1] > y2) {
return ;
}
if(a[p].Min[0] >= x1 && a[p].Max[0] <= x2 && a[p].Max[1] <= y2 && a[p].Min[1] >= y11) {
a[p].c = newc ;
a[p].mark = newc ;
return ;
}
down(p) ;
if(a[p].d[0] >= x1 && a[p].d[0] <= x2 && a[p].d[1] >= y11 && a[p].d[1] <= y2) {
a[p].c = newc ;
}
if(a[p].l) {
upda(a[p].l , x1, y11 , x2 , y2 , newc) ;
}
if(a[p].r) {
upda(a[p].r , x1, y11 , x2 , y2 , newc) ;
}
}
L ans ;
void ask(L p , L x, L y) {
if(a[p].d[0] == x && a[p].d[1] == y) {
ans = a[p].c ;
return ;
}
if(a[p].mark != 0) {
ans = a[p].mark ;
return ;
}
down(p) ;
if(a[p].l) {
L ll = a[p].l ;
if(x >= a[ll].Min[0] && x <= a[ll].Max[0] && y >= a[ll].Min[1] && y <= a[ll].Max[1]) {
ask(ll , x , y) ;
}
}
if(a[p].r) {
L ll = a[p].r ;
if(x >= a[ll].Min[0] && x <= a[ll].Max[0] && y >= a[ll].Min[1] && y <= a[ll].Max[1]) {
ask(ll , x , y) ;
}
}
}
int main(){
L t ;
scanf("%lld",&t);
while(t-- ){
scanf("%lld%lld%lld",&n,&c,&q) ;
for(L i = 1; i <= n ; i ++ )vec[i].clear() ;
for(L i = 2; i <= n ; i ++ ){
L fa ;
scanf("%lld",&fa) ;
vec[fa].push_back(i) ;
}
cnt = 0 ;
dfs(1,1) ;
for(L i = 1; i <= n; i ++ ){
a[i].l = a[i].r = 0;
a[i].d[0] = l[i] ;
a[i].d[1] = deep[i] ;
}
root = build(1,n,0) ;
L sum = 0;
for(L i = 1; i <= q; i ++ ){
L aa , ll , cc ;
scanf("%lld%lld%lld",&aa,&ll,&cc);
L x1 = l[aa] ;
L x2 = r[aa] ;
L y11 = deep[aa] ;
L y2 = deep[aa] + ll ;
if(cc != 0) {
upda(root , x1 , y11 , x2 , y2 , cc) ;
}
else {
ans = -1 ;
L x = l[aa] ;
L y = deep[aa] ;
ask(root, x , y) ;
sum += ans * i ;
sum %= mod ;
}
}
printf("%lld\n",sum) ;
}
}