-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathlongest path in a tree using dfs.cpp
113 lines (75 loc) · 1.89 KB
/
longest path in a tree using dfs.cpp
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define ll long long int
#define mod 1000000007
#define for1(k,n) for(ll i=k;i<n;i++)
#define for2(k,n) for(ll j=k;j<n;j++)
#define E cout<<endl
vector <int> *adj;
int *vis;
int *dist;
void dfs(int s)
{
if (vis[s] == 0)
vis[s] = 1;
for (auto i : adj[s])
if (vis[i] == 0)
{ dist[i] = dist[s] + 1;
dfs(i);
dist[i] = dist[s] + 1;
}
}
int main()
{
//////////////////////////////////////start...............
int n, e, u, v;
cin >> n ;
e = n - 1;
adj = new vector <int> [n + 1];
vis = new int [n + 1];
dist = new int [n + 1];
//fill(vis, vis + n + 1, 0);
fill(vis, vis + n + 1, 0);
fill(dist, dist + n + 1, -1);
for1(0, e)
{
cin >> u >> v;
adj[u].pb(v);
adj[v].pb(u);
}
dist[1] = 0;
dfs(1);
int max = 0; int enode;
for (int i = 1; i <= n; i++)
{ if (dist[i] > max)
{ enode = i;
max = dist[i];
}
}
fill(dist, dist + n + 1, -1);
fill(vis, vis + n + 1, 0);
dist[enode] = 0;
dfs(enode);
max = -1;
for (int i = 1; i <= n; i++)
{ if (dist[i] > max)
max = dist[i];
}
cout << max;
/////////////////////////////end................................... ....
return 0;
}
//c v a s selecting text or x for selecting cut
//ctrl+d after selecting text to select same type
//ctrl+shift+d for copy and paste the line below it
//ctrl+del to delete a text
//ctrl+left to jump left of line or vice versa
//ctrl+shift+"/" to comment whole block and vice versa for undo
//ctrl+"/" for commenting a line
/*
when N <= 10, then both O(N!) and O(2N) are ok (for 2N probably N <= 20 is ok too)
when N <= 100, then O(N3) is ok (I guess that N4 is also ok, but never tried)
when N <= 1.000, then N2 is also ok
when N <= 1.000.000, then O(N) is fine (I guess that 10.000.000 is fine too, but I never tried in contest)
finally when N = 1.000.000.000 then O(N) is NOT ok, you have to find something better…*/