简单算法的python实现