码迷,mamicode.com
首页 > 其他好文 > 详细

[bzoj2783][JLOI2012]树_树的遍历

时间:2018-08-26 14:50:02      阅读:184      评论:0      收藏:0      [点我收藏+]

标签:git   ras   注释   jin   stream   data   ios   fine   oid   

树 bzoj2783 JLOI2012

题目大意:给定一棵n个点的树。求满足条件的路径条数。说一个路径是满足条件的,当且仅当这条路径上每个节点深度依次递增且点权和为S。

注释:$1\le n\le 10^5$,$1\le S,val_i\le 10^3$。


想法:翻lijinnn的blog翻到的水题。

我们直接遍历整棵树,遍历的时候维护全局桶。然后在回溯的时候将这个点对应的dis删除。这样遍历到每个点时桶内对应的就是这个点到根节点的dis桶,直接统计答案即可。

最后,附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#define N 100010
using namespace std;
multiset<int>s;
bool v[N];
int to[N],nxt[N],head[N];
int w[N];
int tot,sum;
int root;
int ans=0;
inline char nc()
{
	static char *p1,*p2,buf[100000];
	return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
int read()
{
	int x=0; char c=nc();
	while(!isdigit(c)) c=nc();
	while(isdigit(c)) x=(x<<3)+(x<<1)+c-‘0‘,c=nc();
	return x;
}
inline void add(int x,int y)
{
	to[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
}
void dfs(int x,int pre,int data)
{
	s.insert(data);
	int tmp=data+w[x];
	ans+=s.count(tmp-sum);
	for(int i=head[x];i;i=nxt[i])
		dfs(to[i],x,tmp);
	s.erase(data);
}
int main()
{
	int n=read(); sum=read();
	for(int i=1;i<=n;i++) w[i]=read();
	for(int x,y,i=1;i<n;i++)
	{
		x=read(),y=read();
		add(x,y); v[y]=1;
	}
	for(int i=1;i<=n;i++) if(!v[i]) root=i;
	dfs(root,0,0);
	printf("%d",ans);
	return 0;
}

小结:裙子的简单数据结构的例题。

[bzoj2783][JLOI2012]树_树的遍历

标签:git   ras   注释   jin   stream   data   ios   fine   oid   

原文地址:https://www.cnblogs.com/ShuraK/p/9537232.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!