螺旋矩阵 IV

/*
 * 题目思路：
 * 1. 初始化一个 m x n 大小的矩阵，初始值为 -1。
 * 2. 使用 IntStream 来生成螺旋顺序的索引。
 * 3. 使用方向数组和 lambda 表达式来控制方向变化。
 * 4. 遍历链表，将节点的值填入矩阵中。
 * 5. 当遇到边界或已经填充过的位置时，改变方向。
 */

import java.util.stream.IntStream;
import java.util.Arrays;

class ListNode {
    int val;
    ListNode next;
    ListNode(int x) { val = x; }
}

public class SpiralMatrixStream {
    public int[][] spiralMatrix(int m, int n, ListNode head) {
        int[][] matrix = new int[m][n];
        IntStream.range(0, m).forEach(i -> Arrays.fill(matrix[i], -1));
        int[] dx = {0, 1, 0, -1};
        int[] dy = {1, 0, -1, 0};
        int[] pos = {0, 0};
        int[] dir = {0};
        ListNode curr = head;

        IntStream.range(0, m * n).forEach(i -> {
            if (curr != null) {
                matrix[pos[0]][pos[1]] = curr.val;
                curr = curr.next;
            }
            int nx = pos[0] + dx[dir[0]];
            int ny = pos[1] + dy[dir[0]];
            if (nx < 0 || nx >= m || ny < 0 || ny >= n || matrix[nx][ny] != -1) {
                dir[0] = (dir[0] + 1) % 4;
                nx = pos[0] + dx[dir[0]];
                ny = pos[1] + dy[dir[0]];
            }
            pos[0] = nx;
            pos[1] = ny;
        });
        return matrix;
    }
}