洛谷 P3252 [JLOI2012] 树

---

读题就读趋势了，还以为是每个深度都可以选一个，然后深度升序就可以了，以为是个按深度的01背包。

但是前面还说了是一条路径，路径是不能断开的。那就从每个点开始爆搜一次就好了。

看了一下范围n<=1e5，n^2爆搜理论上是不行的，但是这道题实在是淼。正解看dfs下方。

伪AC代码（dfs）

#include <bits/stdc++.h>
using namespace std;

inline int read(){
    int x=0;char c=getchar();
    while(c<48 or c>57)c=getchar();
    while(c>=48 and c<=57)x=(x<<3)+(x<<1)+(c xor 48),c=getchar();
    return x;
}

const int N=1e5+5;
int n,s,w[N],ans;//有多少节点权值深度为s
vector<int>e[N];

void dfs(int now,int sum){
    if(sum>s)return;
    if(sum==s){
        ans++;
        return;
    }
    for(auto i:e[now])dfs(i,sum+w[i]);
}
int main(){
    ios::sync_with_stdio(false);
    n=read(),s=read();
    for(int i=1;i<=n;++i)w[i]=read();
    for(int i=1,x,y;i<=n-1;++i){
        x=read(),y=read();
        e[x].push_back(y);
    }

    for(int i=1;i<=n;++i)
        dfs(i,w[i]);

    cout<<ans<<endl;
    return 0;
}

思路

因为是树，所以每个节点仅有1个父节点，存下每个节点到根节点的前缀和，以倍增的形式往上跳，把所有小于<s的都跳了，如果等于s就路径+1。

每个点倍增一次，复杂度nlogn，此题正解！

代码

无，爆搜能过还要什么倍增（暴论）