python简单算法