SSL2675 2017年8月9日提高组T3 难题（dfs）

2017年8月9日提高组T3 难题

Description

小C得到了一棵树，这棵树每个点都有一个权值且1为根节点。无聊的小C又随机了一个权值s，现在他想知道这棵树上有多少条路径的节点权值总和恰好为s，且满足该路径中节点的深度必须是升序的。

Input

第一行是两个正整数n,s，接下来一行n个正整数，表示每个节点的权值。
接下来n-1行，每行包含两个正整数，表示树上的一条边。

Output

输出一个数，表示满足条件的路径数。

分析：先以1为根建树，然后dfs每条路径，如果当前权值和大于s则直接退出即可。（据说正解是树上前缀和+二分）

代码

#include <cstdio>
#define maxn 300000
using namespace std;

struct arr
{
    int x,y,next;
}a[maxn];

struct edge
{
    int x,y,next;
}b[maxn];

int n,m,ans,lsa[maxn],lsb[maxn],c[maxn],l,la;
bool v[maxn];

void add(int u,int v)
{
    l++;
    b[l].x=u;
    b[l].y=v;
    b[l].next=lsb[u];
    lsb[u]=l;
}

void dfs1(int o)
{
    v[o]=true;
    for (int i=lsb[o];i;i=b[i].next)
    if (!v[b[i].y])
    {
        la++;
        a[la].x=o;
        a[la].y=b[i].y;
        a[la].next=lsa[o];
        lsa[o]=la;
        dfs1(b[i].y);
    }
}

void dfs(int o,int s)
{
    if (s>m) return;
    if (s==m) 
    {
        ans++;
        return;
    }
    for (int i=lsa[o];i;i=a[i].next)
        dfs(a[i].y,s+c[a[i].y]);
}

int main()
{
//  freopen("diff.in","r",stdin);
//  freopen("diff.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&c[i]);
    for (int i=1;i<n;i++)
    {
        int p,q;
        scanf("%d%d",&p,&q);
        add(p,q);
        add(q,p);
    }
    dfs1(1);
    for (int i=1;i<=n;i++)
        dfs(i,c[i]);
    printf("%d",ans);
    fclose(stdin);
    fclose(stdout);
}