题目描述:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-dM1R7HPn-1607157484643)(C:\Users\Stuedent\AppData\Roaming\Typora\typora-user-images\image-20201127211106195.png)]
Solution
一、 O ( n 2 ) O(n^2) O(n2)暴力
枚举每一个点,扫一遍她的子树,合法的点累加进答案。
二、 O ( n 2 ) O(n^2) O(n2)暴力
我们换一个角度来打暴力
考虑每一个点对父亲的贡献。
每次向上找父亲,累加答案。
三、可能的正解
我们尝试优化二算法。
设 s u m [ i ] sum[i] sum[i]表示根节点到i的路径上的边权和
我们假设当前节点为 x x x,找到一个深度最小的点 y y y使得 s u m [ x ] − s u m [ y ] < = a [ x ] sum[x]-sum[y]<=a[x] sum[x]−sum[y]<=a[x]
显然的,x能对x->y路径上的所有点都产生1的贡献。
对于二叙述的方案,我们找父亲的情况可以用树上倍增来进行优化。
对于x->y路径上区间加1的操作,我们可以用树链剖分或者数据结构来维护。
时间复杂度 O ( n l o g 2 n ) O(nlog^2n) O(nlog2n)
可能吧。。
四、 99.99999999 % 99.99999999\% 99.99999999%的正解
我们发现上述区间累加的东西其实可以不用数据结构来实现,还有一个东西能维护区间累加操作
就是树上差分
差分的性质:对差分数组做一遍前缀和操作就是原数组基础上的增量。
每次树上倍增找到合法的(x,y)之后,我们只需要令 f a [ x ] fa[x] fa[x]的答案+1, f a [ y ] fa[y] fa[y]的答案-1.
然后回溯的时候把答案往上传递一下即可(相当于求一遍前缀和)
Code
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 2e5+100;
int n,a[N];
struct Node{
int y,Next,v;
}e[2*N];
int len , linkk[N];
int fa[N][30] , sum[N];
int Ans[N];
void Insert(int x,int y,int v){
e[++len] = (Node){y,linkk[x],v};
linkk[x] = len;
}
void dfs(int x,int faa,int v){
fa[x][0] = faa; sum[x] = v;
for (int i = linkk[x]; i; i = e[i].Next){
int y = e[i].y; if (y == faa) continue;
dfs(y,x,v+e[i].v);
}
}
void find_fa(){
for (int j = 1; j < 30; j++)
for (int i = 1; i <= n; i++)
fa[i][j] = fa[fa[i][j-1]][j-1];
}
void Dfs(int x,int faa){
int Fa = 0;
int X = x;
for (int i = 29; i >= 0; i--)
if (fa[X][i] && sum[x]-sum[fa[X][i]]<=a[x]) X = fa[X][i];
Ans[fa[x][0]]++; Ans[fa[X][0]]--;
for (int i = linkk[x]; i; i = e[i].Next){
int y = e[i].y; if (y == faa) continue;
Dfs(y,x);
Ans[x]+=Ans[y];
}
}
signed main(){
scanf("%lld",&n);
for (int i = 1; i <= n; i++) scanf("%lld",&a[i]);
for (int i = 2,y,v; i <= n; i++)
scanf("%lld %lld",&y,&v) , Insert(i,y,v) , Insert(y,i,v);
dfs(1,0,0);
find_fa();
Dfs(1,0);
for (int i = 1; i <= n; i++) printf("%lld ",Ans[i]);
return 0;
}
总结:
面对树上亦或是区间的区间修改问题,我们通常可以用数据结构来维护。而特殊的,有时差分也可以,我们不能把它忘记。
特殊的,如果对一个节点的子树进行操作,我们可以用dfs序将子树的问题转化成区间问题而后用数据结构维护。