参照云姐的代码,简单修该