Code for the mit google interview hack