第二题也挺恶心的，改造第一题代码