【SSLGZ 2675】2017年8月9日提高组T3 难题

问题描述
小C得到了一棵树，这棵树每个点都有一个权值且1为根节点。无聊的小C又随机了一个权值s，现在他想知道这棵树上有多少条路径的节点权值总和恰好为s，且满足该路径中节点的深度必须是升序的。
输入
第一行是两个正整数n,s，接下来一行n个正整数，表示每个节点的权值。
接下来n-1行，每行包含两个正整数，表示树上的一条边。
输出
输出一个数，表示满足条件的路径数。
样例输入
3 3
1 2 3
1 2
1 3
样例输出
2
算法讨论
这里用的是暴力水解……搞个邻接表顺着它向上找，要是加起来的权值大于答案则退出换路找……数据水可以玄学AC……正解应该用树上倍增。

#include<cstdio>
using namespace std;
#define maxn 100007
struct edge
{
    int f,t,n;
}a[maxn];
int p[maxn],ls[maxn];
int n,s,ans,x;

int dfs(int dep,int t,int ss)
{
    if (ss==s)
    {
        ans+=1;
        return 0;
    }
    if (ss>s)
        return 0;
    if (dep!=1)
        x=a[ls[x]].t;
    if (x!=0)
        dfs(dep+1,x,ss+p[x]);
    return 0;
}

int main()
{
    scanf("%d%d",&n,&s);
    for (int i=1;i<=n;i++)
        scanf("%d",&p[i]);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&a[i].f,&a[i].t);
        a[i].n=ls[a[i].f];
        ls[a[i].f]=i;
    }
    for (int i=1;i<=n;i++)
    {
        x=a[ls[i]].t;
        dfs(1,i,p[i]);
    }
    printf("%d",ans);
    return 0;
}

Pixiv ID：33746695