2783: [JLOI2012]树
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 588 Solved: 347
Description
第一行是两个整数N和S,其中N是树的节点数。
第二行是N个正整数,第i个整数表示节点i的正整数。
接下来的N-1行每行是2个整数x和y,表示y是x的儿子。
输出格式:
输出路径节点总和为S的路径数量。
输入样例: |
输出样例: |
3 3 1 2 3 1 2 1 3 |
2 |
数据范围:
对于30%数据,N≤100;
对于60%数据,N≤1000;
对于100%数据,N≤100000,所有权值以及S都不超过1000。
这个是JLOI2012的T1,发出来仅为了试题完整
=============================================================================================
在这个问题中,给定一个值S和一棵树。在树的每个节点有一个正整数,问有多少条路径的节点总和达到S。路径中节点的深度必须是升序的。假设节点1是根节点,根的深度是0,它的儿子节点的深度为1。路径不必一定从根节点开始。
Input
第一行是两个整数N和S,其中N是树的节点数。
第二行是N个正整数,第i个整数表示节点i的正整数。
接下来的N-1行每行是2个整数x和y,表示y是x的儿子。
Output
输出路径节点总和为S的路径数量。
Sample Input
1 2 3
1 2
1 3
Sample Output
HINT
对于100%数据,N≤100000,所有权值以及S都不超过1000。
题解
邻接表存图,dfs查找路径,set记录可能的路径s' 值
代码
#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <queue>
#include <typeinfo>
#include <map>
#include <stack>
typedef long long ll;
using namespace std;
inline ll read()
{
ll x=,f=;
char ch=getchar();
while(ch<''||ch>'')
{
if(ch=='-')f=-;
ch=getchar();
}
while(ch>=''&&ch<='')
{
x=x*+ch-'';
ch=getchar();
}
return x*f;
}
//************************************************************************************** set<int > s;
int t;
int n,sum;
struct ss
{
int to,next;
}e[];
int sss;
int k,ans;
int head[];
int a[];
void init()
{
t=;
memset(head,,sizeof(head));
}
void add(int u,int v)
{
e[t].to=v;
e[t].next=head[u];
head[u]=t++;
}
void dfs(int x,int sum)
{
sum+=a[x];
if(s.count(sum-sss))ans++;
s.insert(sum);
for(int i=head[x];i;i=e[i].next)
{
dfs(e[i].to,sum);
}
s.erase(sum);
}
int hash[];
int main()
{
scanf("%d%d",&n,&sss);
init();
for(int i=;i<=n;i++)
{
scanf("%d",&a[i]);
}
int x,y;
for(int i=;i<n;i++)
{
scanf("%d%d",&x,&y);
add(x,y);
hash[y]=;
}
for(int i=;i<=n;i++)
if(!hash[i]){k=i;break;}
s.clear();
s.insert();
ans=;
sum=;
dfs(k,); cout<<ans<<endl;
return ;
}