读题就读趋势了,还以为是每个深度都可以选一个,然后深度升序就可以了,以为是个按深度的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,此题正解!
代码
无,爆搜能过还要什么倍增(暴论)