Create 1008(2).py

a more simple way to realize the goal