Evo mog rešenja
Prepravio sam par sitnih grešaka na koje sam naleteo dok sam pregledao kod.
Dodao sam i 10-ak test primere koje sam koristio.