Codeforces 600E Lomsat gelral 树上启发式合并

我要吐槽一下.
为什么这题又看不了AC代码,又看不了数据,洛谷上面交还UKE?

题意

给一棵1为根的树,每个点有个颜色,求每一个点的子树里出现最多的颜色的和.

题解

我们用两组$n$个$map\ cnt和sum$分别存储每一个点的子树每一个颜色出现的次数和每一个点的子树每一个出现次数的颜色和.
那么预处理的时候c=read(),cnt[i][c]=1,sum[i][1]=c;
接下来我们以1为根节点进行一波$dfs$.
每一次我们扫一个节点的所有儿子的时候,我们保存它的重儿子(子树大小最大的儿子)的数据,将其它儿子的数据合并到重儿子的数据上.
每一次合并的时候,$map$的大小会至少变成原来的两倍,而最多会变$log(n)$次,所以最后的复杂度就是$O(n\times log^2(n))$.

#include<bits/stdc++.h> //Ithea Myse Valgulious
namespace chtholly{
typedef long long ll;
#define re0 register int
#define rec register char
#define rel register ll
#define gc getchar
#define pc putchar
#define p32 pc(' ')
#define pl puts("")
/*By Citrus*/
inline int read(){
  int x=0,f=1;char c=gc();
  for (;!isdigit(c);c=gc()) f^=c=='-';
  for (;isdigit(c);c=gc()) x=(x<<3)+(x<<1)+(c^'0');
  return f?x:-x;
  }
template <typename mitsuha>
inline bool read(mitsuha &x){
  x=0;int f=1;char c=gc();
  for (;!isdigit(c)&&~c;c=gc()) f^=c=='-';
  if (!~c) return 0;
  for (;isdigit(c);c=gc()) x=(x<<3)+(x<<1)+(c^'0');
  return x=f?x:-x,1;
  }
template <typename mitsuha>
inline int write(mitsuha x){
  if (!x) return 0&pc(48);
  if (x<0) x=-x,pc('-');
  int bit[20],i,p=0;
  for (;x;x/=10) bit[++p]=x%10;
  for (i=p;i;--i) pc(bit[i]+48);
  return 0;
  }
inline char fuhao(){
  char c=gc();
  for (;isspace(c);c=gc());
  return c;
  }
}using namespace chtholly;
using namespace std;
const int yuzu=1e5;
typedef ll fuko[yuzu|10];
typedef map<int,ll> kyoko[yuzu|10];
vector<int> lj[yuzu|10];
kyoko cnt,sum;
fuko llx;

void dfs(int u,int fa){
for (int v:lj[u]) if (v^fa){
  dfs(v,u);
  if (cnt[u].size()<cnt[v].size()){
    swap(cnt[u],cnt[v]);
    swap(sum[v],sum[u]);
    }
  /*将u的重儿子的map交换到u的位置.*/
  for (auto p:cnt[v]){
    sum[u][cnt[u][p.first]]-=p.first;
    /*u这个位置的p.first这个颜色的数量不再是cnt[u][p.first]了,要减掉.*/
    cnt[u][p.first]+=p.second;
    /*重新增加之.*/
    sum[u][cnt[u][p.first]]+=p.first;
    /*合并加回去.*/
    }
  }
llx[u]=sum[u].rbegin()->second;
}

int main(){
int i,n=read(),x,y;
for (i=1;i<=n;++i) sum[i][cnt[i][x=read()]=1]=x;
for (i=1;i<n;++i){
  x=read(),y=read();
  lj[x].push_back(y);
  lj[y].push_back(x);
  }dfs(1,0);
for (i=1;i<=n;++i) write(llx[i]),p32;
}

谢谢大家.