I'd propose 2 solutions: one using recursion; and a second one a bit harder to read but more efficient (for huge amount of data recursion is very consuming). If it's a home work, must be unit tested!