有神能贴一下算法岗的py代码吗，想学习一下。。。