Hello codeforces,

Recently I was practicing 1800 rated problems and I came across this latter picking problem. I implemented O(n ** 2) solution and submitted in pypy. here is the link to the submission (pypy submission) But it got tle at test 5. I was very confused why this is happening as the n for this problem is 2000.

I changed the interpreter to python 3 and submitted the solution and it gets passed.

Can anybody tell me what is the potential problem here?

here is the python3 submission (python 3 submission).

both submissions are identical but I got tle on pypy submission.