给定一个值
S
和一棵树。在树的每个节点有一个正整数,问有多少条路径的节点总和为
我们可以维护这样一个数据结构,他的前半部分是一个保留原来信息的队列
后半部分是一个栈
DFS一遍,当深搜到一个点时将这个点加入队列,同时队头向后调整,使队列中元素之和
<=s
<script id="MathJax-Element-145" type="math/tex"><=s</script> ,等于
s
记录
当一个点出栈时将队尾删除,同时队头向前调整,使队列中元素之和刚好
<=s
<script id="MathJax-Element-148" type="math/tex"><=s</script>
这算是栈和队列的结合使用了……
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#define N 100000+5
#define M 200000+5
using namespace std;
int n,r,h,s;
int a[N],q[N],fa[N];
int head[N];
struct graph
{
int next, to;
graph() {}
graph(int _next, int _to)
: next(_next), to(_to) {}
} edge[M];
inline void add(int x, int y)
{
static int cnt = 0;
edge[++cnt] = graph(head[x], y);
head[x] = cnt;
edge[++cnt] = graph(head[y], x);
head[y] = cnt;
}
int sum;
int ans ;
void DFS(int x)
{
q[++r] = a[x];
sum += a[x];
while(sum>s)
sum-=q[++h];
if(sum==s)++ans;
for(int i=head[x];i;i=edge[i].next)
if(edge[i].to!=fa[x])
{
fa[edge[i].to] = x;
DFS(edge[i].to);
}
sum-=q[r--];
while(h && sum + q[h]<=s)
sum += q[h--];
}
inline int read()
{
int x=0,f=1;char ch = getchar();
while(ch < '0' || ch > '9'){if(ch == '-')f=-1;ch = getchar();}
while(ch >='0' && ch <='9'){x=(x<<1)+(x<<3)+ch-'0';ch = getchar();}
return x*f;
}
int main()
{
cin >> n >> s;
for(int i=1;i<=n;++i)
a[i] = read();
for(int i=1;i<n;++i)
{
int x = read() , y = read();
add(x,y);
}
DFS(1);
cout<<ans<<endl;
}