Cod sursa(job #3238109)

Utilizator Anul2024Anul2024 Anul2024 Data 19 iulie 2024 15:21:49
Problema Diametrul unui arbore Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.07 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin ("darb.in");
ofstream fout ("darb.out");
int max1[100001],max2[100001],dp[100001];
int n,i,x,y;
vector <int> v[100001];
void dfs (int nod,int t)
{
    bool ok=0;
    for (int i=0; i<v[nod].size (); i++)
    {
        int vecin=v[nod][i];
        if (vecin!=t)
        {
            ok=1;
            dfs (vecin,nod);
            if (max1[vecin]+1>=max1[nod])
            {
                max2[nod]=max1[nod];
                max1[nod]=max1[vecin]+1;
            }
            else if (max1[vecin]+1>max2[nod])
                max2[nod]=max1[vecin]+1;
            dp[nod]=max (dp[nod],dp[vecin]);
        }
    }
    if (!ok)
        max1[nod]=1;
    dp[nod]=max (dp[nod],max1[nod]+max2[nod]-1);
}
int main ()
{
    ///dp[i]=lg. diam. pt subarb. lui i
    ///dp[i]=max (dp[vecin],lanturi maxime din nodul i)
    fin>>n;
    for (i=1; i<=n; i++)
    {
        fin>>x>>y;
        v[x].push_back (y);
        v[y].push_back (x);
    }
    dfs (1,0);
    fout<<dp[1];
    return 0;
}