2783: [JLOI2012]树
Time Limit: 1 Sec Memory Limit: 128 MBDescription
数列
提交文件：sequence.pas/c/cpp
输入文件：sequence.in
输出文件：sequence.out
问题描述：
把一个正整数分成一列连续的正整数之和。这个数列必须包含至少两个正整数。你需要求出这个数列的最小长度。如果这个数列不存在则输出-1。
输入格式：
每行包含一个正整数n。
每个文件包含多行，读入直到文件结束。
输出格式：
对于每个n，输出一行，为这个数列的最小长度。
第一行是两个整数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。
数据范围：
对于所有数据，n≤2
63。
这个是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
3 3
1 2 3
1 2
1 3
1 2 3
1 2
1 3
Sample Output
2
HINT
对于100%数据，N≤100000，所有权值以及S都不超过1000。
这题应该不难，若x是y的祖先，令sum[i][j]表示以i为根，向上走2^j步经过的所有点的权值和。对于每一个节点，若m≥sum[i][j]就往上跳，并让m减去该值。如果跳完权值刚好为0，则ans+1。
Code：
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<set>
#include<cmath>
#include<algorithm>
#define inf 2000000000
#define N 100005
using namespace std;
int n,s,cnt,ans;
int v[N],last[N],sum[N];
multiset<int> q;
struct edge{int to,next;}e[N];
void insert(int u,int v)
{
e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}
void dfs(int x)
{
if(q.find(sum[x]-s)!=q.end())ans++;
q.insert(sum[x]);
for(int i=last[x];i;i=e[i].next)
{
sum[e[i].to]=sum[x]+v[e[i].to];
dfs(e[i].to);
}
q.erase(q.find(sum[x]));
}
int main()
{
scanf("%d%d",&n,&s);
q.insert(0);
for(int i=1;i<=n;i++)scanf("%d",&v[i]);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
insert(u,v);
}
sum[1]=v[1];dfs(1);
printf("%d\n",ans);
return 0;
}