Pergunta de entrevista da empresa TuSimple

Traversing an NxN matrix question.