BZOJ2783 [JLOI2012]树（洛谷P3252）

版权声明：蒟蒻Blog随意转载 https://blog.csdn.net/a1799342217/article/details/81515284

倍增

BZOJ题目传送门
洛谷题目传送门

水水更健康

可以发现以$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;
}