TAK, masz!
Tok rozumowania jest całkowicie poprawny [ale możnaby go jeszcze "lekko podrasować" - czytaj przyśpieszyć - ale to już inna bajka]. Natomiast kod już nie jest dobry i nawet komentarze nic tu nie zmienią. W tym wypadku, pomagający mają w małym paluszku cały algorytm i najczęściej taki miejscami nadmiarowy miejscami błędny, miejscami za długi komentarz tylko zaciemniają [na mój gust] kod.
jbc, ideone po prawej stronie, na środku, małymi literami napisał to
Kiepsko szukałeś - czytaj testowałeś. Np zobacz:
https://ideone.com/uN4S9A38