【bzoj2783】【JLOI2012】树

题意

  有一棵n个节点的树，每个节点有一个权值Wi
  一条路径的权值为∑Wu，u为路径上的点（包括端点）
  求权值为S的，从上往下的路径条数

解法

DFS：
  从每一个节点往下进行DFS，记录累计权值sum，当sum≥S的时候就返回，如果sum=S，就对答案产生贡献

复杂度

O（nlen），len为所求路径的平均长度，一般在【10，20】之间

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
using namespace std;
const int MAXN=100010;
struct node
{
    int next,to;
}t[MAXN];
int head[MAXN],num;
int w[MAXN];
int n,s,ans;
void add(int u,int v)
{
    t[++num]=(node){ head[u],v };
    head[u]=num;
}
void dfs(int k,int sum)
{
    if( sum>=s )   { ans+=(sum==s);return ; }
    for(int i=head[k],x; i ;i=t[i].next)
    {
        x=t[i].to;
        dfs( x,sum+w[x] );
    }
}
int main()
{
    int u,v;
    scanf("%d%d",&n,&s);
    for(int i=1;i<=n;i++)   scanf("%d",&w[i]);
    for(int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&u,&v);
        add( u,v );
    }
    for(int i=1;i<=n;i++)   dfs( i,w[i] );
    printf("%d\n",ans);
    return 0;
}