solution for 001 in go
002 in go