2783: [JLOI2012]树

树上前缀和。。。。。。

每次访问一个节点的时候查询是否存在祖先的前缀和与该节点的前缀和之差为S，若有则答案+1，因为前缀和之差为一段树链的和，就酱。

用set维护一下就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
using namespace std;
const int N=100000+5;
multiset<int>s;
struct Edge{int to,next;}e[N*2];
int head[N],cnt,v[N],S,sum[N],ans;
void ins(int u,int v){
	e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;
}
void dfs(int u){
	if(s.find(sum[u]-S)!=s.end())ans++;
	s.insert(sum[u]);
	for(int i=head[u];i;i=e[i].next){
		sum[e[i].to]=sum[u]+v[e[i].to];
		dfs(e[i].to);
	}
	s.erase(s.find(sum[u]));
}
int main(){
	int n;scanf("%d%d",&n,&S);
	s.insert(0);
	for(int i=1;i<=n;i++)scanf("%d",&v[i]);
	int from,to;
	for(int i=1;i<n;i++){
		scanf("%d%d",&from,&to);
		ins(from,to);
	}sum[1]=v[1];
	dfs(1);
	printf("%d",ans);
	return 0;
}