【bzoj2783】【JLOI2012】【树】【set】

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

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

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

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

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

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

题解：处理一下每个点到根的路径权值和，动态的维护一个set，每dfs到一个点询问一下set里有没有当前点到根的路径权值和-s的点。有的话答案+1即可。

#include<iostream>
#include<cstdio>
#include<set>
#define N 100010
using namespace std;
set<int>s;
int n,m,point[N],next[N*2],k,a[N],u,v,cnt,ans,deep[N];
struct use{
	int st,en;
}e[2*N];
void add(int x,int y)
{
	next[++cnt]=point[x];point[x]=cnt;
	e[cnt].st=x;e[cnt].en=y;
}
void dfs(int x,int fa)
{
	for (int i=point[x];i;i=next[i])
	 	if (e[i].en!=fa)
	 	{
		 deep[e[i].en]=deep[x]+a[e[i].en];
		 s.insert(deep[e[i].en]);
	 	 if (s.find(deep[e[i].en]-k)!=s.end()) ans++;
	 	 dfs(e[i].en,x);
	     s.erase(deep[e[i].en]);
	    }
}
int main()
{
	scanf("%d%d",&n,&k);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	for (int i=1;i<=n-1;i++)
	 {
	   scanf("%d%d",&u,&v);
	   add(u,v);add(v,u);
	 }
	s.insert(0);deep[1]=a[1];s.insert(deep[1]);dfs(1,0);
	cout<<ans<<endl;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。