看一下我这个代码有没有问题。

你这里循环中都没用到n，可以直接删掉n>=3