第二题用python7行就搞定了，java比较麻烦