角谷猜想python代码