I believe this implementation is too complicated for interview answer. Have a look at mine above which is simple and straightforward.