minor changes but made the code run 33% faster

[xfgavin]

加了一些边界条件的处理，可以让代码执行速度提高33%。