Recently I interviewed at AWS at amazon for a SDE position and got an offer too. I used Python for all my DS and Algo problems. It purely depends on your comfort level with the language.