Cod sursa(job #2040847)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 16 octombrie 2017 17:00:12
Problema Zvon Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.39 kb
/*
I love you Rose
I hope you love me too
*/
#include<bits/stdc++.h>
using namespace std;
ifstream f("zvon.in");
ofstream g("zvon.out");
int t,a,b;
int n,ar[100003];
bool wm[100003];
vector<int>v[100003];
vector<int>ar2;
void dfs1(int nod)
{
    ar[nod]=0;
    if(v[nod].size()==0){
        ar[nod]=1;
        return;
    }
    for(int i=0;i<v[nod].size();++i)
    {
        dfs1(v[nod][i]);
        ar[nod]+=ar[v[nod][i]];
    }
}
int dfs2()
{
    int t=0;
    int d=1;
    while(d<n)
    {
        ++t;
        int prvs=ar2.size();
        for(int i=0;i<prvs;++i)
        {
            int max1=0;
            int who=0;
            int d1=v[ar2[i]].size();
            for(int j=0;j<d1;++j)
                if(wm[v[ar2[i]][j]]==0)
                    if(ar[v[ar2[i]][j]]>max1)
                        max1=ar[v[ar2[i]][j]],who=v[ar2[i]][j];
            if(who!=0){
                ++d;
                ar2.push_back(who);
                wm[who]=1;
            }
        }
    }
    return t;
}
int main()
{
    f>>t;
    for(;t;--t)
    {
        f>>n;
        for(int i=1;i<=n;++i)
            v[i].clear();
        memset(wm,0,100002);
        ar2.clear();
        for(int i=1;i<n;++i)
        {
            f>>a>>b;
            v[a].push_back(b);
        }
        dfs1(1);
        ar2.push_back(1);
        g<<dfs2()<<'\n';
    }
    return 0;
}