倍增
水水更健康
可以发现以 i i 为起点的一条合法路径一定是在到根的路径上的。那么预处理后直接倍增跳一跳就好了。
代码:
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define F inline
using namespace std;
int n,k,S,ans,dep[N],fa[N][18],s[N][18];
F char readc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
return l==r?EOF:*l++;
}
F int _read(){
int x=0; char ch=readc();
while (!isdigit(ch)) ch=readc();
while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
return x;
}
F void Make(){
for (int j=1;j<18;j++)
for (int i=1;i<=n;i++){
fa[i][j]=fa[fa[i][j-1]][j-1];
s[i][j]=s[i][j-1]+s[fa[i][j-1]][j-1];
}
}
F int LCA(int x){
int ret=0;
for (int j=17;~j;j--)
if (s[x][j]+ret<=S)
ret+=s[x][j],x=fa[x][j];
return ret==S;
}
int main(){
n=_read(),S=_read();
for (int i=1;i<=n;i++) s[i][0]=_read();
for (int i=1,x,y;i<n;i++)
x=_read(),y=_read(),fa[y][0]=x;
Make();
for (int i=1;i<=n;i++) ans+=LCA(i);
return printf("%d\n",ans),0;
}