Palindromes in a Tree CodeForces - 914E

https://vjudge.net/problem/CodeForces-914E

点分就没一道不卡常的？

卡常记录：

1.把不知道为什么设的(unordered_map)s换成了(int[])s

2.减少一次cal2和clr

 #pragma GCC optimize("Ofast")
 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 using namespace std;
 #define fi first
 #define se second
 #define mp make_pair
 #define pb push_back
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pi;
 struct E
 {
     int to,nxt;
 }e[];
 int f1[],ne;
 int sz[],fx[],d[];ll ans[],ta,an1[];
 char ss[];int a[];
 bool vis[];
 int root,sum;
 int lft[];
 int s[];
 void getroot(int x,int fa)
 {
     sz[x]=;fx[x]=;
     for(int k=f1[x];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             getroot(e[k].to,x);
             sz[x]+=sz[e[k].to];
             fx[x]=max(fx[x],sz[e[k].to]);
         }
     fx[x]=max(fx[x],sum-sz[x]);
     if(fx[x]<fx[root])    root=x;
 }
 void getsz(int x,int fa)
 {
     sz[x]=;
     for(int k=f1[x];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             getsz(e[k].to,x);
             sz[x]+=sz[e[k].to];
         }
 }
 void calc(int u,int fa)
 {
     an1[u]+=s[d[u]^a[root]],ta+=s[d[u]^a[root]];
     for(int i=;i<;i++)    an1[u]+=s[d[u]^a[root]^lft[i]],ta+=s[d[u]^a[root]^lft[i]];
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
             calc(e[k].to,u);
 }
 void addd(int u,int fa)
 {
     s[d[u]]++;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             d[e[k].to]=d[u]^a[e[k].to];
             addd(e[k].to,u);
         }
 }
 void deld(int u,int fa)
 {
     s[d[u]]--;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
             deld(e[k].to,u);
 }
 void cal2(int u,int fa)
 {
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             cal2(e[k].to,u);
             an1[u]+=an1[e[k].to];
         }
     ans[u]+=an1[u];
 }
 void clr(int u,int fa)
 {
     an1[u]=;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
             clr(e[k].to,u);
 }
 void solve(int u)
 {
     d[u]=;vis[u]=;ta=;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             d[e[k].to]=a[e[k].to];
             addd(e[k].to,u);
         }
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             clr(e[k].to,u);
             deld(e[k].to,u);
             calc(e[k].to,u);
             addd(e[k].to,u);
         }
     ans[u]+=ta/;ta=;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
             deld(e[k].to,u);
     s[]++;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             calc(e[k].to,u);
             cal2(e[k].to,u);
         }
     s[]--;ans[u]+=ta;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             root=;getsz(e[k].to,);sum=sz[e[k].to];
             getroot(e[k].to,);solve(root);
         }
 }
 int n;
 int main()
 {
     fx[]=0x3f3f3f3f;
     int i,x,y;
     lft[]=;
     for(i=;i<=;i++)    lft[i]=lft[i-]<<;
     scanf("%d",&n);
     for(i=;i<n;i++)
     {
         scanf("%d%d",&x,&y);
         e[++ne].to=y;e[ne].nxt=f1[x];f1[x]=ne;
         e[++ne].to=x;e[ne].nxt=f1[y];f1[y]=ne;
     }
     scanf("%s",ss+);
     for(i=;i<=n;i++)    a[i]=lft[ss[i]-'a'];
     root=;sum=n;getroot(,);
     solve(root);
     for(i=;i<=n;i++)    printf("%lld ",ans[i]+);
     return ;
 }