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