[bzoj2783] 树

 

题意：给出一个有根树，点上有点权，求点权和为s的路径数，路径上的点的深度要求递增

 

题解：

这题算比较水了，在树上用set维护一下权值和就可以了

用set维护的好处就是，可以实时维护路径和（加入一个值后可以很方便的删除），不会重复更新答案，之前写了个预处理前缀和的，会算重额

 

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<set>
#define ll long long
#define N 100010
using namespace std;

int n,s,e_num,ans;
int nxt[N*2],to[N*2],h[N],sum[N],val[N];

set<int> st;

int gi() {
  int x=0,o=1; char ch=getchar();
  while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
  if(ch=='-') o=-1,ch=getchar();
  while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
  return o*x;
}

void add(int x, int y) {
  nxt[++e_num]=h[x],to[e_num]=y,h[x]=e_num;
}

void dfs(int u) {
  if(st.find(sum[u]-s)!=st.end()) ans++;
  st.insert(sum[u]);
  for(int i=h[u]; i; i=nxt[i]) {
    int v=to[i];
    sum[v]=sum[u]+val[v];
    dfs(v);
  }
  st.erase(st.find(sum[u]));
}

int main() {
  n=gi(),s=gi();
  for(int i=1; i<=n; i++) val[i]=gi();
  for(int i=1; i<n; i++) {
    int x=gi(),y=gi();
    add(x,y);
  }
  st.insert(0),sum[1]=val[1];
  dfs(1);
  printf("%d", ans);
  return 0;
}