N-gram Dictionary — Interview Query python problem