[JLOI 2012]树

Description

       在这个问题中，给定一个值S和一棵树。在树的每个节点有一个正整数，问有多少条路径的节点总和达到S。路径中节点的深度必须是升序的。假设节点1是根节点，根的深度是0，它的儿子节点的深度为1。路径不必一定从根节点开始。

Input

       第一行是两个整数N和S，其中N是树的节点数。

       第二行是N个正整数，第i个整数表示节点i的正整数。

       接下来的N-1行每行是2个整数x和y，表示y是x的儿子。

Output

       输出路径节点总和为S的路径数量。

Sample Input

3 3
1 2 3
1 2
1 3

Sample Output

2

HINT

对于100%数据，N≤100000，所有权值以及S都不超过1000。

题解

遍历一遍，将根到当前节点路径上的所有的点的树上前缀和压入栈中。

查找时二分栈即可。

//It is made by Awson on 2017.9.29
#include <set>
#include <map>
#include <cmath>
#include <ctime>
#include <queue>
#include <stack>
#include <vector>
#include <cstdio>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define LL long long
#define Max(a, b) ((a) > (b) ? (a) : (b))
#define Min(a, b) ((a) < (b) ? (a) : (b))
#define sqr(x) ((x)*(x))
#define lowbit(x) ((x)&(-(x)))
using namespace std;
const int N = 100000;
void read(int &x) {
    char ch; bool flag = 0;
    for (ch = getchar(); !isdigit(ch) && ((flag |= (ch == '-')) || 1); ch = getchar());
    for (x = 0; isdigit(ch); x = (x<<1)+(x<<3)+ch-48, ch = getchar());
    x *= 1-2*flag;
}

int n, s, u, v;
int val[N+5];
int sum[N+5], top;
struct tt {
    int to, next;
}edge[N+5];
int path[N+5], tot;
int ans;

void add(int u, int v) {
    edge[++tot].to = v;
    edge[tot].next = path[u];
    path[u] = tot;
}
bool erge(int l, int r) {
    while (l <= r) {
        int mid = (l+r)>>1;
        if (sum[top]-sum[mid] == s) return true;
        if (sum[top]-sum[mid] > s) l = mid+1;
        else r = mid-1;
    }
    return false;
}
void dfs(int u) {
    sum[++top] = sum[top-1]+val[u];
    ans += erge(0, top);
    for (int i = path[u]; i; i = edge[i].next)
        dfs(edge[i].to);
    top--;
}
void work() {
    read(n), read(s);
    for (int i = 1; i <= n; i++) read(val[i]);
    for (int i = 1; i < n; i++)
        read(u), read(v), add(u, v);
    dfs(1);
    printf("%d
", ans);
}
int main() {
    work();
    return 0;
}