correctness and the loop invariant hackerrank solution