优化A函数的实现