cses tree solution-distinct colours

#include<bits/stdc++.h>
using namespace std;
//problem link
//https://cses.fi/problemset/task/1139
//please merge this as hacktoberfest-accepted label
//this problem uses very good use of pointer .
//hope u like this
//thanks
 
 
#define ll long long
#define ld long double
#define rep(i,a,b) for(ll i=a;i<b;i++)
#define repb(i,a,b) for(ll i=a;i>=b;i--)
#define err() cout<<"--------------------------"<<endl; 
#define errA(A) for(auto i:A)   cout<<i<<" ";cout<<endl;
#define err1(a) cout<<#a<<" "<<a<<endl
 
 
 
#define err2(a,b) cout<<#a<<" "<<a<<" "<<#b<<" "<<b<<endl
#define err3(a,b,c) cout<<#a<<" "<<a<<" "<<#b<<" "<<b<<" "<<#c<<" "<<c<<endl
#define err4(a,b,c,d) cout<<#a<<" "<<a<<" "<<#b<<" "<<b<<" "<<#c<<" "<<c<<" "<<#d<<" "<<d<<endl
#define pb push_back
#define all(A)  A.begin(),A.end()
#define allr(A)    A.rbegin(),A.rend()
#define ft first
#define sd second
 
#define pii pair<int,int>
#define pll pair<ll,ll>
#define V vector<ll>
#define S set<ll>
#define VV vector<V>
#define Vpll vector<pll>
 
#define endl "\n"

ll binaryexp(ll a,ll b,ll i)
{
    if(b==0)
        return 1LL;
    if(b==1)
        return a;
    ll k=binaryexp(a,b/2,i);
    if(b&1)
    {
        return (((k*k)%i)*a)%i;
    }
    else
        return (k*k)%i;
}
 
V col(200005);
set<int> *st[200005];
V ans(200005);
void dfs(VV &a,int i,int p)
{
    int mxs=-1,mxi=-1;
    int n=a.size();
    for(auto x:a[i])
    {
        if(x!=p)
        {
            dfs(a,x,i);
            int jj=st[x]->size();
            if(jj>mxs)
            {
                mxs=st[x]->size();
                mxi=x;
            }
        }
    }
    // err1(mxi);
    if(mxi==-1)
    {
        st[i]=new set<int>();
        st[i]->insert(col[i]);
    }
    else
    {
        st[i]=st[mxi];
        for(auto x:a[i])
        {
            if(x!=p && x!=mxi)
            {
                for(auto kk: *st[x])
                st[i]->insert(kk);
            }
        }
        st[i]->insert(col[i]);
    }
    ans[i]=st[i]->size();
 
}
int main()
    {
        ios_base::sync_with_stdio(false);
        cin.tie(NULL); cout.tie(NULL);
        #ifndef ONLINE_JUDGE
         clock_t tStart = clock();
        freopen("input.txt","r",stdin);
        freopen("output.txt","w",stdout);
        #endif
        int t;
        // cin>>t;
        t=1;
           
        while(t--)
        {
            int n;
            cin>>n;
            VV a(n);
            rep(i,0,n)
            {
                cin>>col[i];
            }
            rep(i,0,n-1)
            {
                int x,y;
                cin>>x>>y;
                x-=1;
                y-=1;
                a[x].pb(y);
                a[y].pb(x);
            }
            dfs(a,0,-1);
            rep(i,0,n)
            {
                cout<<ans[i]<<" ";
            }
 
 
        }
        #ifndef ONLINE_JUDGE
            printf("\nRun Time -> %.10fs\n", (double)(clock()-tStart) / CLOCKS_PER_SEC);
            #endif
}