BZOJ3696 化合物

我们可以树形dp...

令f[p][d]表示以p为根的子树，与p距离为d的结点数

然后我们计算答案：

一种是从某个节点q到根p的方案，对和为d的贡献是1

另一种是p的一个子树中的节点x到另一个子树中的节点y的方案，对和为d[x] ^ d[y]的贡献为1

第二种我们可以通过f暴力求出，话说什么时候去研究一下FWT啊。。。

/**************************************************************
    Problem: 3696
    User: rausen
    Language: C++
    Result: Accepted
    Time:476 ms
    Memory:207460 kb
****************************************************************/
 
#include <cstdio>
#include <algorithm>
 
using namespace std;
const int N = 1e5 + 5;
const int H = 525;
 
struct edge {
    int next, to;
    edge() {}
    edge(int _n, int _t) : next(_n), to(_t) {}
} e[N];
 
int n;
int first[N], tot;
int f[N][H], ans[H], dep[N];
 
inline int read() {
    int x = 0, sgn = 1;
    char ch = getchar();
    while (ch < '0' || '9' < ch) {
        if (ch == '-') sgn = -1;
        ch = getchar();
    }
    while ('0' <= ch && ch <= '9') {
        x = x * 10 + ch - '0';
        ch = getchar();
    }
    return sgn * x;
}
 
inline void Add_Edge(int x, int y) {
    e[++tot] = edge(first[x], y), first[x] = tot;
}
 
#define y e[x].to
void dfs(int p) {
    int x, i, j;
    f[p][0] = 1;
    for (x = first[p]; x; x = e[x].next) {
        dfs(y);
        for (i = 0; i <= dep[p]; ++i)
            for (j = 0; j <= dep[y]; ++j)
                ans[i ^ (j + 1)] += f[p][i] * f[y][j];
        dep[p] = max(dep[p], dep[y] + 1);
        for (i = 0; i <= dep[y]; ++i)
            f[p][i + 1] += f[y][i];
    }
}
#undef y
 
int main() {
    int i, mx;
    n = read();
    for (i = 2; i <= n; ++i)
        Add_Edge(read(), i);
    dfs(1);
    for (mx = H - 1; mx; --mx)
        if (ans[mx]) break;
    for (i = 0; i <= mx; ++i)
        printf("%d
", ans[i]);
    return 0;
}