我代码是这样子的，感觉和你的思路差不多