标签：Educational int ll dsu Codeforces cnt 31 col maxx

题目链接

题意:给你一棵以1为根n个点的树，问你以i为根的子树的众数和是多少

思路:dsu是一种优化暴力的手段 首先进行轻重链剖分 然后只记录重链的信息 轻链的信息就直接暴力查找 经过证明这样复杂度可以是nlogn。

#include<bits/stdc++.h>
#define ll long long int
using namespace std;
inline ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
inline ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
int moth[13]={0,31,28,31,30,31,30,31,31,30,31,30,31};
int dir[4][2]={1,0 ,0,1 ,-1,0 ,0,-1};
int dirs[8][2]={1,0 ,0,1 ,-1,0 ,0,-1, -1,-1 ,-1,1 ,1,-1 ,1,1};
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
const int N=100007;
vector<int> G[N];
int col[N],dp[N],son[N];
ll cnt[N],ans[N];
ll maxx=-inf,sum=0;
int po;
void dfs(int u,int fa){
    dp[u]=1;
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
        dp[u]+=dp[v];
        if(dp[v]>dp[son[u]]) son[u]=v; //找重儿子 
    }
}
void add(int u,int fa,int w){
    cnt[col[u]]+=w;
    if(cnt[col[u]]>maxx) maxx=cnt[col[u]],sum=col[u];
    else if(cnt[col[u]]==maxx) sum+=col[u];
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(v==fa||v==po) continue;
        add(v,u,w);
    }
}
void dfss(int u,int fa,int opt){
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(v==fa) continue;
        if(son[u]!=v) dfss(v,u,0); //优先遍历轻儿子 
    }
    if(son[u]) dfss(son[u],u,1),po=son[u];//po是标记当重链的重儿子 在算当前节点时可以不用遍历重儿子 
    add(u,fa,1);
    po=0;
    ans[u]=sum;
    if(!opt) add(u,fa,-1),maxx=-inf,sum=0; //消除轻儿子的影响 
}
int main(){
    ios::sync_with_stdio(false);
    int n; cin>>n;
    for(int i=1;i<=n;i++){
        cin>>col[i];
    }
    for(int i=1;i<=n-1;i++){
        int u,v; cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0); dfss(1,0,1);
    for(int i=1;i<=n;i++)
        cout<<ans[i]<<" ";
    return 0;
}

标签：Educational,int,ll,dsu,Codeforces,cnt,31,col,maxx
来源： https://www.cnblogs.com/wmj6/p/11127301.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。