补充下代码实例

多考虑两种情况，并补充了下代码实例，这都有点像单元测试了。。