2783: [JLOI2012]树( dfs + BST )

直接DFS, 然后用set维护一下就好了.... O(nlogn)

--------------------------------------------------------------------------------

#include<bits/stdc++.h>

 

#define rep(i, n) for(int i = 0; i < n; ++i)

#define clr(x, c) memset(x, c, sizeof(x))

#define foreach(i, x) for(__typeof(x.begin()) i = x.begin(); i != x.end(); i++)

 

using namespace std;

 

const int maxn = 100009;

 

struct edge {

int to;

edge* next;

} E[maxn], *pt = E, *head[maxn];

  

inline void addedge(int u, int v) {

pt->to = v, pt->next = head[u];

head[u] = pt++;

}

 

int w[maxn], n, s, root, ans = 0;

bool F[maxn];

set<int> S;

 

void dfs(int x, int d) {

d += w[x];

if(S.find(d - s) != S.end()) ans++;

S.insert(d);

for(edge* e = head[x]; e; e = e->next)

   dfs(e->to, d);

S.erase(d);

}

 

int main() {

freopen("test.in", "r", stdin);

clr(F, 0), clr(head, 0);

cin >> n >> s;

rep(i, n) scanf("%d", w + i);

rep(i, n - 1) {

int x, y;

scanf("%d%d", &x, &y); x--, y--;

addedge(x, y);

F[y] = true;

}

S.clear(); S.insert(0);

rep(i, n) if(!F[i])

   dfs(i, 0);

cout << ans << "\n";

return 0;

}

-------------------------------------------------------------------------------- 

2783: [JLOI2012]树

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 572   Solved: 342
[ Submit][ Status][ Discuss]

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。

Source