Amazon interview question

Simple stack question, with an optimization to reduce run time.