洛谷3252(bzoj2783)[JLOI2012]树 前缀和+单调队列

题目

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

题解

路径的和就相当于数列中一个区间的和，于是又用上了前缀和。。
由于节点的数都为正整数，所以前缀和一定单调递增，那么就像单调队列一样，甚至根本不用维护其单调性。。。
那么遍历一遍这个树，设s[i]表示前缀和，假如树中有值为s[i]-s的节点，那么ans++

代码

#include <cstdio>

using namespace std;

int n,s,ans;
int a[100005];
int ls[100005],ne[200005],y[200005];
long long b[100005];

bool find(int r,long long k){
    int l=0,ss=0;
    while (l<=r){
        int mid=(l+r)/2;
        if (b[mid]<=k) ss=mid,l=mid+1;
                  else r=mid-1;
    }
    if (b[ss]==k) return true;
    return false;
}

void dfs(int x,int k){
    b[k]=b[k-1]+a[x];
    if (find(k,b[k]-s)) ans++;
    for (int i=ls[x];i;i=ne[i])
    {
        dfs(y[i],k+1);
    }
    b[k]=0;
}

int main(){
    scanf("%d%d",&n,&s);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for (int i=1;i<n;i++){
        int x,to;
        scanf("%d%d",&x,&to);
        ne[i]=ls[x];ls[x]=i;y[i]=to;
    }
    b[0]=0;
    dfs(1,1);
    printf("%d\n",ans);
    return 0; 
}