
读题就读趋势了,还以为是每个深度都可以选一个,然后深度升序就可以了,以为是个按深度的01背包。
但是前面还说了是一条路径,路径是不能断开的。那就从每个点开始爆搜一次就好了。
看了一下范围n<=1e5,n^2爆搜理论上是不行的,但是这道题实在是淼。正解看dfs下方。
伪AC代码(dfs)
- #include
- 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<
- return 0;
- }
思路
因为是树,所以每个节点仅有1个父节点,存下每个节点到根节点的前缀和,以倍增的形式往上跳,把所有小于
每个点倍增一次,复杂度nlogn,此题正解!
代码
无,爆搜能过还要什么倍增(暴论)