python实现算法的代码