python实现算法改进