Competitive-Programming/diameter of tree.cpp at master · potter1024/Competitive-Programming · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
    ll n;
    cin>>n;
    for(ll i=0;i<n-1;i++)
    {
        ll a,b;
        cin>>a>>b;
        adj[a].pb(b);
        adj[b].pb(a);
    }
    queue<ll> q1;
    q1.push(1);
    vector<ll> dist(N,0),vis(N,0);
    vis[1]=1;
    ll end,ma=0;
    while(!q1.empty())
    {
        ll x=q1.front();
        q1.pop();
        for(ll i=0;i<adj[x].size();i++)
        {
            if(!vis[adj[x][i]])
            {
                q1.push(adj[x][i]);
                vis[adj[x][i]]=1;
                dist[adj[x][i]]=dist[x]+1;
                if(dist[adj[x][i]]>ma)
                {
                    ma=dist[adj[x][i]];
                    end=adj[x][i];
                }
            }
        }
    }
    for(ll i=1;i<=n;i++)
    {
        vis[i]=0;
        dist[i]=0;
    }
    q1.push(end);
    vis[end]=1;
    ma=0;
    while(!q1.empty())
    {
        ll x=q1.front();
        q1.pop();
        for(ll i=0;i<adj[x].size();i++)
        {
            if(!vis[adj[x][i]])
            {
                q1.push(adj[x][i]);
                vis[adj[x][i]]=1;
                dist[adj[x][i]]=dist[x]+1;
                if(dist[adj[x][i]]>ma)
                {
                    ma=dist[adj[x][i]];
                    end=adj[x][i];
                }
            }
        }
    }
    vector<ll> diam;
    diam.pb(end);
    ll u=end,d=dist[end];
    while(d)
    {
        for(ll i=0;i<adj[u].size();i++)
        {
            if(dist[adj[u][i]]==d-1)
            {
                diam.pb(adj[u][i]);
                d--;
                u=adj[u][i];
                break;
            }
        }
    }

    // diam will store the path of diameter