Задача: 1424. Diagonal Traverse II
Сложность: medium

Дан двумерный целочисленный массив nums, верните все элементы nums в диагональном порядке.

Пример:

Input: nums = [[1,2,3,4,5],[6,7],[8],[9,10,11],[12,13,14,15,16]]
Output: [1,6,2,8,7,3,9,4,12,10,5,13,11,14,15,16]

👨‍💻 Алгоритм:

1⃣Инициализируйте очередь с (0, 0) и список ответов ans.

2⃣Пока очередь не пуста:
Извлеките (row, col) из очереди.
Добавьте nums[row][col] в ans.
Если col == 0 и row + 1 в пределах массива, добавьте (row + 1, col) в очередь.
Если col + 1 в пределах текущей строки, добавьте (row, col + 1) в очередь.

3⃣Верните ans.

😎 Решение:

class Solution {
    function findDiagonalOrder($nums) {
        $queue = [[0, 0]];
        $ans = [];
        
        while (!empty($queue)) {
            list($row, $col) = array_shift($queue);
            $ans[] = $nums[$row][$col];
            
            if ($col === 0 && $row + 1 < count($nums)) {
                $queue[] = [$row + 1, $col];
            }
            
            if ($col + 1 < count($nums[$row])) {
                $queue[] = [$row, $col + 1];
            }
        }
        
        return $ans;
    }
}

Ставь 👍 и забирай 📚 Базу знаний