Codechef TREEPATH 线段树优化dp+dsu on tree

题意

给定一棵无根树，每个节点上都写了一个整数。
你的任务就是统计有多少种方法可以将这棵树分解为若干条路径，使得每个节点恰好属于一条路径，而且每条路径的节点上的数字之和非负。
$1 ≤ 每组数据中的 N 之和 ≤ 10^5$
$1 ≤ N ≤ 10^5$
$−10^4 ≤ Ai ≤ 10^4$

分析

首先来考虑暴力要怎么打。
先随便定一个根，设$f[i]$表示以$i$为根的子树被分解成若干条路径且每条路径均满足题目条件的方案，$val[i]$表示从$i$到根路径上节点的权值和。
因为节点$i$一定要属于其中一条路径，那么该路径的两个端点必然满足其中一个是$i$或在不同子树内。
假如确定了两个端点，分别为$x,y$，那么该路径合法的条件是