-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmaximum sum of all subtrees.cp
139 lines (132 loc) · 4.35 KB
/
maximum sum of all subtrees.cp
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
//#include <bits/stdc++.h>
#include <iostream>
#include <climits>
#include <vector>
#include <cstdlib>
#include <iomanip>
#include <algorithm>
#include <fstream>
#include <cmath>
#include <math.h>
#include <set>
#include <time.h>
#include <map>
#include <fstream>
#include <cstdio>
#include <vector>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <algorithm>
#include <array>
#include <bitset>
#include <complex>
#include <numeric>
#include <unordered_map>
using namespace std;
#define sz(a) int((a).size())
#define ll long long
#define pb push_back
#define ld long double
#define pi pair<ll,ll>
#define mp make_pair
#define vi vector<ll>
#define vii vector<pi>
#define endl "\n"
#define ff first
#define ss second
#define all(c) (c).begin(),(c).end()
#define allr(c) (c).rbegin(),(c).rend()
#define rep(i,n) for(ll i=0;i<n;i++)
#define cpresent(c,x) (find(all(c),x) != (c).end())
#define input(a,n) for(long long i=0;i<n;i++)cin>>a[i]
#define output(a,n) for(long long i=0;i<n;i++)cout<<a[i]<<" "
#define inputt(a,n,m) for(ll i=0;i<n;i++)for(ll j=0;j<m;j++)cin>>a[i][j]
#define outputt(a,n,m) for(ll i=0;i<n;i++){for(ll j=0;j<m;j++)cout<<a[i][j]<<" ";cout<<endl;}
#define reset(a,n) memset(a,n,sizeof(a)) // n can only be 0 or -1
#define vl LLONG_MIN
#define vm LLONG_MAX
#define ep 1e-10
#define seed 13331
#define inf 1e10
#define N 1000000
#define minp vector<ll>, greater<ll>
#define M 1000000007
#define MM 1000000009
#define pie 3.1415926535897932384626433832795
#define eu 2.718281828459045
#define she_taught_me_to_code cin.tie(0);cout.tie(0);ios_base::sync_with_stdio(false);cin.tie(NULL); clock_t startTime=clock();
#define time cout<<(double(clock()-startTime )/(double)CLOCKS_PER_SEC)*1000<<" ms"<<endl;
#define debug(k) cout<<"\t-> "<<#k<<" = "<<k<<endl;
/************************************Debugging Steps*****************************************/
// If aplying ceil,floor,pow,etc convert it into integer (might make you eat shit if you don't
// Whenever using stacks,queues,etc always check that their top / front / back elements are not accessed when they are empty, thus causing runtime error
// Think twice before using greedy
// Remember losing 1 or 2 minutes is better than a penalty of 10 minutes
// Use log2 instead of log()/log(2) to avoid shitty errors
// If MLE occurs, try declaring the large array in int instead of long long
// Always initialise anything while declaring, which will not be taken as input
// In case of a problem having decimal calculations, try declaring everything in double rather than typecasting in each step
// always declare the array with n+5 elements to avoid unexpected errors
// Every problem cannot be solved by DFS easily, think of BFS too
/********************************************************************************************/
/********************************************************************************************/
// Your code need to originate from the deepest of your intellectual mind
// Best of Luck, may there never be "Wrong answer on test ##"
bool sortbysec(const pair<ll,ll> &a,const pair<ll,ll> &b)
{
return (a.second < b.second);
}
ll gcd(ll a,ll b)
{
return b?gcd(b, a % b):a;
}
ll min(ll a,ll b)
{
return a<b?a:b;
}
ll max(ll a,ll b)
{
return a>b?a:b;
}
ll n;
vi adj[N],ch(N);
void dfs(ll u,ll last)
{
ch[u]=1;
for(ll v:adj[u])
if(v!=last)
dfs(v,u),ch[u]+=ch[v];
}
ll ans=0;
void _dfs(ll u,ll last,ll s)
{
ans=max(ans,s);
for(ll v:adj[u])
if(v!=last)
_dfs(v,u,s+n-2*ch[v]);
}
int32_t main()
{
she_taught_me_to_code
// freopen("input.txt","r",stdin);
// freopen("output.txt","w",stdout);
cin>>n;
for(ll i=1;i<n;i++)
{
ll a,b;
cin>>a>>b;
adj[a].pb(b);
adj[b].pb(a);
}
ll s=0;
dfs(1,0);
for(ll i=1;i<=n;i++)
s+=ch[i];
_dfs(1,0,s);
cout<<ans<<endl;
}