BZOJ2783: [JLOI2012]树

Description

数列

提交文件：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⁶³。

这个是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

Sample Output

2

HINT

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

 

水题++，set大法好。

#include<cstdio>
#include<cctype>
#include<queue>
#include<cmath>
#include<set>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
const int BufferSize=1<<16;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
    if(head==tail) {
        int l=fread(buffer,1,BufferSize,stdin);
        tail=(head=buffer)+l;
    }
    return *head++;
}
inline int read() {
    int x=0,f=1;char c=Getchar();
    for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=Getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=100010;
int n,m,ans,val[maxn],first[maxn],next[maxn],to[maxn],in[maxn],e;
void AddEdge(int v,int u) {
    to[++e]=v;next[e]=first[u];first[u]=e;in[v]++;
}
set<int> S;
void dfs(int x,int dep) {
    if((*S.lower_bound(dep-m))==dep-m) ans++;
    S.insert(dep);
    ren dfs(to[i],dep+val[to[i]]);
    S.erase(dep);
}
int main() {
    n=read();m=read();
    rep(i,1,n) val[i]=read();
    rep(i,2,n) AddEdge(read(),read());
    S.insert(0);
    rep(i,1,n) if(!in[i]) dfs(i,val[i]);
    printf("%d
",ans);
    return 0;
}