Samsara Machine Learning Intern interview questions

I was given 4 questions. The first 2 were easy peasy. The third one was an implementation-heavt problem involving traveling matrix elements in a certain way. The fourth question was not good and seemed very unfair, solved it but couldn't completely pass it as one of the test cases threw a time complexity error, which I did not have time to go back a rethink my strategy.