Il codice postato risolve il problema in modo preciso, elegante ed efficiente.

Prova a studiarlo con un po' più di attenzione ed a capirlo.