Capital One interview question

CodeSignal medium - Variation of diagonal matrix traverse (LeetCode 498)