思路对了，代码没写全