【dfs】【哈希表】bzoj2783 [JLOI2012]树

因为所有点权都是正的，所以对每个结点u来说，每条从根到它的路径上只有最多一个结点v符合d(u,v)=S。

所以我们可以边dfs边把每个结点的前缀和pre[u]存到一个数据结构里面，同时查询pre[u]-S是否存在。

数据结构用set、hashtable(随便卡)(需要支持删除，由于总是删掉最后一个，因此可以实现)都行。

#include<cstdio>
#include<cstring>
using namespace std;
#define MAXN 100001
#define MOD 100003
struct HashTable
{
    int v[MAXN],next[MAXN],first[MOD],en;
    HashTable(){memset(first,-1,sizeof(first));en=0;}
    void Insert(const int &x)
    {int o=x%MOD; v[en]=x; next[en]=first[o]; first[o]=en++;}
    void Delete(const int &x) {first[x%MOD]=next[--en];}
    bool Find(const int &x)
    {int o=x%MOD; for(int i=first[o];i!=-1;i=next[i]) if(v[i]==x) return 1; return 0;}
}T;
int n,K,w[MAXN],ans;
int v[MAXN],first[MAXN],next[MAXN],en;
void AddEdge(const int &U,const int &V){v[++en]=V; next[en]=first[U]; first[U]=en;}
void dfs(int U,int W)
{
    if(K<=W) ans+=T.Find(W-K);
    T.Insert(W);
    for(int i=first[U];i;i=next[i])
      dfs(v[i],W+w[v[i]]);
    T.Delete(W);
}
int main()
{
    int a,b;
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;++i) scanf("%d",&w[i]);
    for(int i=1;i<n;++i) {scanf("%d%d",&a,&b); AddEdge(a,b);}
    T.Insert(0); dfs(1,w[1]); printf("%d\n",ans);
    return 0;
}