颓红警

事实证明，小可爱是最可爱的嘤！ ——佚名
现在小可爱在颓游戏，但是他遇到了一个问题：
小可爱率领的部队现在面对的是敌军在这一地区的驻军，敌国战争机器的运作很大程度上依赖指挥，所以敌军内部是严明分级的，就是说，全部敌军可以看作一棵树，每只敌军部队（树上每个节点）有其战斗力。你可以对任意敌军部队发动进攻，小可爱的部队有战斗力p，意味着他的每次进攻将使得被进攻的这支部队的战斗力减少p，对上级指挥系统的打击同时会影响其下级部队。具体来说，当他对点i发动进攻，部队i的战力减少p的同时，对于其子树内点j，部队j的战力减少Max(0,p−dis(i,j)2)（dis(i,j)表示点i,j间简单路径的长度）。如果某支部队战力小于0，那么这支部队就被消灭了，一支部队被消灭不会改变敌军编制（即这棵树的结构不会改变）。
小可爱想知道，你的部队最少发动几次进攻，才能全歼敌军
由于小可爱还要爆手速发展自己实力，所以把这个问题交给了你。
小可爱因为太可爱了，所以受到了一些限制——只有在一个部队的祖先节点都被歼灭之后才能发动进攻去打它，否则它就会被这个部队的祖先节点攻击，这是他不愿意经历的。

1. 树形结构 ‘
2. 只能从上到下进行操作
3. 每个节点值小于0的时候才不可以对其进行操作
4. 1…i - 1中对i会产生影响的，只有令$p-dis[i,j{]}^2>0$ , 也就是如果用深度表示dis , 那么就是$(dis[j]-dis[i]{)}^2<p$ , 就是$dis[j]-dis[i]<\sqrt{p}$, 也就是说对于i来说它的祖先节点只有距离它$\sqrt{p}$以内的才会对i产生影响。
5. 那么也就是说对于i来说，我们只关心距离当前点距离为$\sqrt{p}$的点，大于这个的我们都不关心，这个可以用滑动窗口来维护
6. 那么维护的内容是什么呢。$$\begin{gathered} p-dis[i,j{]}^2 \\ =p-(dis(j)-dis(i){)}^2 \\ =p-dis(i{)}^2-dis(j{)}^2+2\ast dis(i)\ast dis(j) \end{gathered}$$
可以发现对于一个滑动窗口里面，$dis[j]$是需要枚举的，$i$是$j$的祖先节点，那么我们需要维护的也就是区间$dis[i{]}^2,dis[i]$ , 还有$cnt\ast (p-dis[i{]}^2)$ ，这个$cnt$次数,dfs向下传的时候，就相当于区间加一，将当前点的贡献算一下， 同时需要将那一个上面不需要的一个祖先节点的贡献减掉

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <unordered_map>
#include <vector>
#include <map>
#include <list>
#include <queue>
#include <cstring>
#include <cstdlib>
#include <ctime>
#include <cmath>
#include <stack>
#include <set>
#pragma GCC optimize(3 , "Ofast" , "inline")
using namespace std ;
#define ios ios::sync_with_stdio(false) , cin.tie(0) , cout.tie(0)
#define x first
#define y second
#define ls rt << 1
#define rs rt << 1 | 1
typedef long long ll ;
const double esp = 1e-6 , pi = acos(-1) ;
typedef pair<int , int> PII ;
const int N = 1e6 + 10 , INF = 0x3f3f3f3f , mod = 1e9 + 7;
int cinint(){  int t ; scanf("%d" , &t) ;  return t ;}
int cinll(){ll t ;scanf("%lld" , &t) ;return t ;}
ll a[N] , n , p ;
vector<int> v[N] ;
ll size[N] , size1[N] , size2[N] , len , cnt[N] , res[N] , ans = 0 ;
void dfs(int u , int fa , ll d){
  if(d > len) {
    ll t = d - len ;
    size[u] -= cnt[t] ;
    size1[u] -= cnt[t] * t ;
    size2[u] -= cnt[t] * t * t ;
  }
  res[u] -= size[u] * (p - d * d) - size2[u] + 2 * size1[u] * d ;
  ll pos = 0 ;
  if(res[u] >= 0) pos = res[u] / p + 1 , ans += pos ;
  cnt[d] = pos ;
  for(auto x : v[u]) {
    if(x == fa) continue ;
    size[x] = size[u] + pos , size1[x] += size1[u] + pos * d , size2[x] += size2[u] + pos * d * d ;
    dfs(x , u , d + 1) ;
  }
}
int work()
{
  n = cinint() , p = cinint() ;
  len = sqrt(p) + 1 ;
  for(int i = 1; i <= n ;i ++ ) res[i] = cinint() ;
  for(int i = 1; i < n ;i ++ ) {
    int a = cinint() , b = cinint() ;
    v[a].push_back(b) , v[b].push_back(a) ;
  }
  dfs(1 , 0 , 1) ;
  cout << ans << endl ;
  return 0 ;
}
int main()
{

  work() ;
  return 0 ;
}
/*
*/