【题解】CF739B Alyona and a tree

题目描述：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-dM1R7HPn-1607157484643)(C:\Users\Stuedent\AppData\Roaming\Typora\typora-user-images\image-20201127211106195.png)]

---

Solution

一、$O(n^2)$暴力

枚举每一个点，扫一遍她的子树，合法的点累加进答案。

二、$O(n^2)$暴力

我们换一个角度来打暴力

考虑每一个点对父亲的贡献。

每次向上找父亲，累加答案。

三、可能的正解

我们尝试优化二算法。

设$sum[i]$表示根节点到i的路径上的边权和

我们假设当前节点为$x$，找到一个深度最小的点$y$使得$sum[x]-sum[y]<=a[x]$

显然的，x能对x->y路径上的所有点都产生1的贡献。

对于二叙述的方案，我们找父亲的情况可以用树上倍增来进行优化。

对于x->y路径上区间加1的操作，我们可以用树链剖分或者数据结构来维护。

时间复杂度$O(nlo{g}^{2}n)$

可能吧。。

四、$99.99999999\%$的正解

我们发现上述区间累加的东西其实可以不用数据结构来实现，还有一个东西能维护区间累加操作

就是树上差分

差分的性质：对差分数组做一遍前缀和操作就是原数组基础上的增量。

每次树上倍增找到合法的(x,y)之后，我们只需要令$fa[x]$的答案+1，$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序将子树的问题转化成区间问题而后用数据结构维护。