CF600E Lomsat gelral
题目内容
给出一个树,求出每个节点的子树中出现次数最多的颜色的编号和。
思路
那些年,关于板子题的那些事
代码
Dsu on tree:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+10;
int n;
int col[maxn],res[maxn];
struct Edge{
int from,to,nxt;
}e[maxn<<1];
inline int read(){
int x=0;bool fopt=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar())if(ch=='-')fopt=0;
for(;isdigit(ch);ch=getchar())x=(x<<3)+(x<<1)+ch-48;
return fopt?x:-x;
}
int head[maxn],cnt;
inline void add(int u,int v){
e[++cnt].from=u;
e[cnt].to=v;
e[cnt].nxt=head[u];
head[u]=cnt;
}
int dep[maxn],siz[maxn],son[maxn],fa[maxn];
void dfs(int u){
dep[u]=dep[fa[u]]+1;siz[u]=1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa[u])continue;
fa[v]=u;
dfs(v);
siz[u]+=siz[v];
if(!son[u]||siz[son[u]]<siz[v])son[u]=v;
}
}
int Max=-1,sum;
int buc[maxn];
void modify(int u,int val,int pos){
buc[col[u]]+=val;
if(buc[col[u]]>Max){
Max=buc[col[u]];sum=col[u];
}else if(buc[col[u]]==Max)sum+=col[u];
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa[u]||v==pos)continue;
modify(v,val,pos);
}
}
void dsu(int u,bool isson){
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa[u]||v==son[u])continue;
dsu(v,0);
}
if(son[u])dsu(son[u],1);
modify(u,1,son[u]);
res[u]=sum;
if(!isson){
modify(u,-1,0);
sum=Max=0;
}
}
signed main(){
n=read();
for(int i=1;i<=n;i++)
col[i]=read();
for(int i=1;i<n;i++){
int u=read(),v=read();
add(u,v);add(v,u);
}
dfs(1);
dsu(1,0);
for(int i=1;i<=n;i++)
printf("%lld ",res[i]);
return 0;
}
线段树合并:
#include <bits/stdc++.h>
#define int long long
#define lson tr[rt].l
#define rson tr[rt].r
using namespace std;
const int maxn=1e5+10;
int n,tot;
int col[maxn],root[maxn],res[maxn];
struct Edge{
int from,to,nxt;
}e[maxn<<1];
struct SEG{
int l,r,sum,ans;
}tr[maxn*32];
inline int read(){
int x=0;bool fopt=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar())if(ch=='-')fopt=0;
for(;isdigit(ch);ch=getchar())x=(x<<3)+(x<<1)+ch-48;
return fopt?x:-x;
}
int head[maxn],cnt;
inline void add(int u,int v){
e[++cnt].from=u;
e[cnt].to=v;
e[cnt].nxt=head[u];
head[u]=cnt;
}
inline void pushup(int rt){
int u=tr[lson].sum>=tr[rson].sum?lson:rson;
tr[rt].sum=tr[u].sum;
if(tr[lson].sum==tr[rson].sum)
tr[rt].ans=tr[lson].ans+tr[rson].ans;
else tr[rt].ans=tr[u].ans;
}
inline void modify(int &rt,int l,int r,int pos,int v){
if(!rt)rt=++tot;
if(l==r){
tr[rt].sum+=v;
tr[rt].ans=l;
return;
}
int mid=(l+r)>>1;
if(pos<=mid)modify(lson,l,mid,pos,v);
else modify(rson,mid+1,r,pos,v);
pushup(rt);
}
inline int Merge(int a,int b,int l,int r){
if(!a||!b)return a+b;
if(l==r){
tr[a].sum+=tr[b].sum;
tr[a].ans=l;
return a;
}
int mid=(l+r)>>1;
tr[a].l=Merge(tr[a].l,tr[b].l,l,mid);
tr[a].r=Merge(tr[a].r,tr[b].r,mid+1,r);
pushup(a);
return a;
}
void dfs(int u,int fa){
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa)continue;
dfs(v,u);
Merge(root[u],root[v],1,1e5);
}
modify(root[u],1,1e5,col[u],1);
res[u]=tr[root[u]].ans;
}
signed main(){
n=read();tot=n;
for(int i=1;i<=n;i++){
col[i]=read();
root[i]=i;
}
for(int i=1;i<n;i++){
int u=read(),v=read();
add(u,v);add(v,u);
}
dfs(1,0);
for(int i=1;i<=n;i++)
printf("%lld ",res[i]);
return 0;
}