Problem Link: https://codeforces.com/contest/1187/problem/E
I am writing this blog partly because i wanna rethink what the solution is and partly it may help someone. So we are given a graph and we need maximum points by choosing white vertices. Obvious greedy approach to this would be to take the root add points gained by it, then take the children (as this would give the maximum points greedily) and add their points to the contribution and so on. Now the problem is we can't tell which root will be giving us an optimal answer. Also according to the input size given we cant run dfs separately for each vertex. So thus we need an observation to reduce the time complexity so that we can get Accepted. Observation:
Re rooting the vertex of the graph to any of its children only changes few things and this is the main observation needed to solve this problem. Below i choose a graph :
here i will start by choosing 2 as my root. So consider how we can get the max points here. Points by root = N(all the nodes) + points by child 1 + points by child 6.
So we can use dp to get points of children and the answer becomes dp = sz + dp + dp (sz = 9)
Now what all will change if we root the vertex 1 instead of 2 ?
Firstly lets deal with the changes of vertex 2 . Its size changes and it's dp changes so dp -= dp (we remove the left subtree's contribution here) dp -= sz (remember that size also contributed in the inital answer so we need to remove it now) sz -= sz (remove the size finally , dont update this before updating dp due to obv reasons)
Now we have successfully dealt with the changes in the vertex 2 so now we can get the answer with tree rooted at 1 by updating it's previous values as follows dp += dp dp += sz (Because now the size of 1 is no longer just 3 but 9 as its our new root so we update that too.)
This way we can check for each vertex the max points we can get and then finally choose maximum among all of them.
Hope it Helped.
If it did plz upvote :p
Thanks! Please write more tutorial like this of some good problems
Perhaps you should also write a tutorial on how to copy usernames.
I had this username from my game handle (RS6) so i used that only. Why would i copy username from a kid :p
Thanks for the clear explanation. The example graph really helped. Hope you are aware of the other simpler solution using distances from a root, as well.
I really want to thank you for this, you cleared all the doubts that I had regarding this technique. Thank you so much
You explain it very well!
Here are more examples to apply rerooting technique