算法

一道重邮的Python题目，稍微有点复杂（主要是数学部分），记录于此。