CF914E Palindromes in a Tree（点分治）

题面

洛谷
CF

题解

题意：给你一颗 n 个顶点的树（连通无环图）。顶点从 1 到 n 编号，并且每个顶点对应一个在‘a’到‘t’的字母。 树上的一条路径是回文是指至少有一个对应字母的排列为回文。 对于每个顶点，输出通过它的回文路径的数量。 注意：从u到v的路径与从v到u的路径视为相同，只计数一次

性质：回文字符串至多一个字母次数为奇数
因为字母只有'a'~'t'
那么可以状压一下

然后就是套点分治就好了

注意：顶点经过的次数要除以2（因为每条路径算了两次）

Code

#include<bits/stdc++.h>

#define LL long long
#define RG register

using namespace std;
const int N = 200010;
inline int gi() {
    RG int x = 0; RG char c = getchar(); bool f = 0;
    while (c != '-' && (c < '0' || c > '9')) c = getchar();
    if (c == '-') c = getchar(), f = 1;
    while (c >= '0' && c <= '9') x = x*10+c-'0', c = getchar();
    return f ? -x : x;
}

struct node {
    int to, next;
}g[N<<1];
int last[N], gl;
inline void add(int x, int y) {
    g[++gl] = (node) {y, last[x]};
    last[x] = gl;
    return ;
}
int sum, f[N], siz[N], rt;
bool vis[N];
void getroot(int u, int fa) {
    siz[u] = 1; f[u] = 0;
    for (int i = last[u]; i; i = g[i].next) {
        int v = g[i].to; if (v == fa || vis[v]) continue;
        getroot(v, u);
        f[u] = max(f[u], siz[v]);
        siz[u] += siz[v];
    }
    f[u] = max(f[u], sum-siz[u]);
    if (f[rt] > f[u]) rt = u;
    return ;
}
char a[N];
int s[N], t[10050000];
void dfs(int x, int fa, int p, int S) {
    t[S ^= (1 << s[x])] += p;
    for (int i = last[x]; i; i = g[i].next) {
        int v = g[i].to;
        if (v == fa||vis[v]) continue;
        dfs(v, x, p, S);
    }
}
LL ans[N];
LL calc(int x, int fa, int S) {
    S ^= (1 << s[x]);
    LL cnt = t[S];//都为偶数个
    for (int i = 0; i < 20; i++) cnt += t[S^(1<<i)];//出现了一个次数为奇数个
    for (int i = last[x]; i; i = g[i].next) {
        int v = g[i].to;
        if (v == fa || vis[v]) continue;
        cnt += calc(v, x, S);
    }
    ans[x] += cnt;
    return cnt;
}

void solve(int x) {
    vis[x] = 1;
    dfs(x, 0, 1, 0);
    LL cnt = t[0];
    for (int i = 0; i < 20; i++) cnt += t[1<<i];
    //单个一条链
    for (int i = last[x]; i; i = g[i].next) {
        int v = g[i].to; if (vis[v]) continue;
        dfs(v, x, -1, 1<<s[x]);//去掉以v开头的链
        cnt += calc(v, x, 0); //计算组合起来的路径
        dfs(v, x, 1, 1<<s[x]);      
    }
    dfs(x, 0, -1, 0);
    ans[x] += cnt/2;//算了两次啦
    for (int i = last[x]; i; i = g[i].next) {
        int v = g[i].to; if(vis[v]) continue;
        sum = siz[v];rt = 0;
        getroot(v, 0);
        solve(rt);
    }
    return ;
}

int main() {
    //freopen(".in", "r", stdin);
    //freopen(".out", "w", stdout);
    int n = gi();
    for (int i = 1; i < n; i++) {
        int u = gi(), v = gi();
        add(u, v); add(v, u);
    }
    scanf("%s", a);
    for (int i = 0; i < n; i++) s[i+1] = a[i]-'a';
    f[0] = sum = n; rt = 0;
    getroot(1, 0); solve(rt);
    for (int i = 1; i <= n; i++)
        printf("%lld ", ans[i]+1);
    return 0;
}

转载于:https://www.cnblogs.com/zzy2005/p/10183653.html