Amazon interviewers always start with array and then they will ask for more quick solutions with lesser complexity. You can also prepare well for the interview by giving some mock interviews specially designed for Amazon, Flipkart and other product-based companies. Visit moktok.org for more information.