This was my solution with O(n^2) complexity:
https://codeforces.net/contest/1917/submission/238752759
Here is the solution with a lower complexity:
https://codeforces.net/contest/1917/submission/239038045
I did not understand what was done in the solution. Can anyone explain?