I got it wrong, but subsequently I looked in "Cracking the Coding Interview" and there is a question in there that is similar. My advice is to work hard on this book before even applying.