[luoguP3252] [JLOI2012]树(DP)

Posted 蒟蒻zht的博客

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[luoguP3252] [JLOI2012]树(DP)相关的知识,希望对你有一定的参考价值。

传送门

 

树上前缀和。

在树上找一条权值和为 s 的链,其中这个链上的点按深度递增(递减)(不同)

dfs

每搜到一个点求它的前缀和 sum[x],放入 set 中。

在 set 中找 sum[x] - s 的点,如果有,ans++

退出 dfs 的时候再把 sum[x] 从 set 中删除

因为每个点权都是正整数,所以 set 中没有重复元素。

同时也是单调递增,所以简单些不用 set,开个数组再 lower_bound 也行。

 

——代码

技术分享
 1 #include <set>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <iostream>
 5 
 6 const int MAXN = 100001;
 7 int n, s, cnt, ans;
 8 int a[MAXN], head[MAXN], to[MAXN << 1], next[MAXN << 1], f[MAXN], sum[MAXN];
 9 std::set <int> S;
10 
11 inline int read()
12 {
13     int x = 0, f = 1;
14     char ch = getchar();
15     for(; !isdigit(ch); ch = getchar()) if(ch == -) f = -1;
16     for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - 0;
17     return x * f;
18 }
19 
20 inline void add(int x, int y)
21 {
22     to[cnt] = y;
23     next[cnt] = head[x];
24     head[x] = cnt++;
25 }
26 
27 inline void dfs(int u)
28 {
29     sum[u] = sum[f[u]] + a[u];
30     S.insert(sum[u]);
31     if(S.count(sum[u] - s)) ans++;
32     for(int i = head[u]; i ^ -1; i = next[i]) dfs(to[i]);    
33     S.erase(sum[u]);
34 }
35 
36 int main()
37 {
38     int i, x, y;
39     n = read();
40     s = read();
41     memset(head, -1, sizeof(head));
42     for(i = 1; i <= n; i++) a[i] = read();
43     for(i = 1; i < n; i++)
44     {
45         x = read();
46         y = read();
47         f[y] = x;
48         add(x, y);
49     }
50     S.insert(0);
51     dfs(1);
52     printf("%d\n", ans);
53     return 0;
54 }
View Code

 

以上是关于[luoguP3252] [JLOI2012]树(DP)的主要内容,如果未能解决你的问题,请参考以下文章

题解 P3252 [JLOI2012]树

P3252 [JLOI2012]树

洛谷P3252 [JLOI2012]树

bzoj2783: [JLOI2012]树

「JLOI2012」树

Luogu P3258[JLOI2014]松鼠的新家