hdu 5977 Garden of Eden(点分治枚举路径模板)

题意：一棵树，有 n(n≤50000) 个节点，每个点都有一个颜色，共有 k(k≤10) 种颜色，问有多少条路径可以遍历到所有 k 种颜色？（一条路径交换起点终点就算两条哦）

思路：这个点分治很好想到，不过如何判断一条路径上是否包含所有k种颜色是一个问题。但看到k很小，考虑状态压缩，最多有(1<<k)-1种状态，利用点分治可以算出从根节点到所有子节点的路径，或运算可记录经过的路径，将这些路径状态记录，统计每种状态有多少。然后就是计算有多少路径包含k种颜色，枚举数组里每一个数x，若x与某个数或运算后得(1<<k)-1，也就是((1<<k)-1)^x这个数有多少个，为了计算x这条路径上所有的点，这些点与别的路径匹配后包含k种颜色，这些点我们可以通过枚举子集来枚举，然后再与(1<<k)-1进行异或运算，就可以找到了所有的情况。

复杂度

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
int n, k, h[N], cnt, sz[N], rt, vis[N], mx, sn, a[N];
int d[N], haxi[1 << 10];
ll ans;
struct node {
    int v, nt;
} no[N];
void add(int u, int v) {
    no[cnt] = node{v, h[u]};
    h[u] = cnt++;
}
void getroot(int u, int fa) {
    sz[u] = 1;
    int ma = 0;
    for(int i = h[u]; ~i; i = no[i].nt) {
        int v = no[i].v;
        if(!vis[v] && v != fa) {
            getroot(v, u);
            sz[u] += sz[v];
            ma = max(ma, sz[v]);
        }
    }
    ma = max(ma, sn - sz[u]);
    if(mx > ma)
        mx = ma, rt = u;
}
void getsta(int u, int fa, int sta) {
    d[++d[0]] = sta;
    for(int i = h[u]; ~i; i = no[i].nt) {
        int v = no[i].v;
        if(!vis[v] && v != fa)
            getsta(v, u, sta | (1 << a[v]));
    }
}
ll calc(int u, int sta) {
    ll res = 0;
    d[0] = 0;
    memset(haxi, 0, sizeof haxi);
    getsta(u, 0, sta);
    for(int i = 1; i <= d[0]; i++)
        haxi[d[i]]++;
    for(int i = 1; i <= d[0]; i++) {
        haxi[d[i]]--;
        res += haxi[(1 << k) - 1];
        for(int j = d[i]; j; j = (j - 1) & d[i])//枚举子集
            res += haxi[((1 << k) - 1) ^ j];
        haxi[d[i]]++;
    }
    return res;
}
void dfs(int u) {
    vis[u] = 1, ans += calc(u, 1 << a[u]);
    for(int i = h[u]; ~i; i = no[i].nt) {
        int v = no[i].v;
        if(!vis[v])
            ans -= calc(v, (1 << a[u]) | (1 << a[v])), sn = sz[v], rt = 0, mx = 1e9, getroot(v, 0), dfs(rt);
    }
}
int main() {
    while(~scanf("%d%d", &n, &k)) {
        memset(h, -1, sizeof h);
        memset(vis, 0, sizeof vis);
        ans = 0,  cnt = 0;
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]), a[i]--;
        for(int u, v, i = 1; i < n; i++) {
            scanf("%d%d", &u, &v);
            add(u, v), add(v, u);
        }
        sn = n, mx = 1e9, getroot(1, 0), dfs(rt);
        k == 1 ? printf("%lld\n", (ll)n * (ll)n) : printf("%lld\n", ans);
    }
    return 0;
}