在一个网格中可以看到的人数

/*
题目思路:
使用Java Stream API进行实现。我们将网格中的每个位置抽象为一个人，如果该位置有一个人，我们会检查其四个方向上的可见人数。

主要思路与普通Java实现类似，区别在于使用Stream API来遍历和处理数据。
*/

import java.util.stream.IntStream;

public class PeopleVisibilityStream {
    public int countVisiblePeople(int[][] grid) {
        return IntStream.range(0, grid.length)
                .flatMap(i -> IntStream.range(0, grid[i].length)
                        .filter(j -> grid[i][j] == 1)
                        .map(j -> countVisibleInDirection(grid, i, j)))
                .sum();
    }

    private int countVisibleInDirection(int[][] grid, int x, int y) {
        return countDirection(grid, x, y, 0, 1) + // 向右
               countDirection(grid, x, y, 1, 0) + // 向下
               countDirection(grid, x, y, 0, -1) + // 向左
               countDirection(grid, x, y, -1, 0); // 向上
    }

    private int countDirection(int[][] grid, int x, int y, int dx, int dy) {
        int count = 0;
        int i = x + dx;
        int j = y + dy;
        while (i >= 0 && i < grid.length && j >= 0 && j < grid[0].length) {
            if (grid[i][j] == 1) {
                count++;
            }
            i += dx;
            j += dy;
        }
        return count;
    }
}