Domanda di colloquio di Capital One

One matrix question from leetcode