BZOJ 2783 JLOI 2012 树 搜索

做了个测试。

函数调用的耗时的确影响很大。

当dfs(x, fa)这么写的时候是304ms，fa单独开个数组就276ms，写个快速读入208ms，干了加边的函数内联以后是200ms，缓存了双端队列状态以后是176ms（未缓存的代码可以参考popoqqq神的）。Rank 100+ -> 15。不过内存就是蹭蹭蹭地上去了。。68ms的太神了。

可以考虑使用神级输入输出优化的模板，没试过也懒得试了。。

由于路径都是一条链，所以搞个双端队列，维护队列内总和不超过s即可。

开个hash查是否存在s-x_now也是可以的（STL大法好）。

#include <cstdio>
const int N = 100001, M = 200000;
int read() {
    int s = 0, f = 1; char ch = getchar();
    for (; ch < '0' || ch > '9'; ch = getchar()) if (ch == '-') f = -1;
    for (; '0' <= ch && ch <= '9'; ch = getchar()) s = s * 10 + ch - '0';
    return s * f;
}
int p[M], h[N], v[M], cnt = 0, ans = 0, q[N], fa[N], f[N], r[N], sum[N], s, a[N];
void dfs(int x) {
    r[x] = r[fa[x]], f[x] = f[fa[x]], sum[x] = sum[fa[x]];
    q[r[x]++] = a[x]; sum[x] += a[x];
    while (sum[x] > s) sum[x] -= q[f[x]++];
    if (sum[x] == s) ans++;
    for (int i = h[x]; i; i = p[i])
        if (v[i] != fa[x])
            fa[v[i]] = x, dfs(v[i]);
}
int main() {
    int n = read(), i, x, y;
    s = read();
    for (i = 1; i <= n; i++) a[i] = read();
    for (i = 1; i < n; i++) {
        x = read(); y = read();
        p[++cnt] = h[x]; v[cnt] = y; h[x] = cnt;
        p[++cnt] = h[y]; v[cnt] = x; h[y] = cnt;
    }
    dfs(1);
    printf("%d", ans);
    return 0;
}

2783: [JLOI2012]树

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 669  Solved: 390
[ Submit][ Status][ Discuss]

Description

把一个正整数分成一列连续的正整数之和。这个数列必须包含至少两个正整数。你需要求出这个数列的最小长度。如果这个数列不存在则输出-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。