Comment by Tmp1234

Your solution looks better than mine. Is there anything you've gleaned from the leetcode examples I posted?