How can I solve this problem in Python, I found some C# and Java code here: https://sites.google.com/site/tranquangtanqt1990/learn-code/codefights/arcade/2-the-cor
![[愿得一人]](https://www.e-learn.cn/qa/data/avatar/000/00/00/small_000000012.jpg)