力扣 1706. 球会落何处

题目来源：https://leetcode-cn.com/problems/where-will-the-ball-fall/

大致题意：
给一个 m*n 的网格，网格的值为 1 或 -1，为 1 表示该格子有一个从左上角到右下角的隔板，为 -1 表示该格子有一个从右上角到左下角的隔板。
初始时，可以假设在第一行所有格子放下一个球，让球在重力作用下往下掉，如果隔板之间形成了一个 V 或者隔板与墙之间形成了一侧为竖线的“V”，那么球就会卡住
求出最后所有球的位置

大致题意

用一个数组存下所有球的位置，按照规则遍历，模拟出球接下来的位置即可

模拟

球卡住的条件有：

• 当前在左边界，网格值为 -1，与左侧墙构成 V
• 当前在右边界，网格值为 1，与右侧强构成 V
• 当前网格值为 1，右侧网格值为 -1，与右侧网格构成 V
• 当前网格值为 -1，左侧网格值为 1，与左侧网格构成 V

于是可以按如下方法解题：

1. 遍历所有网格
2. 每遍历到一个网格，先判断当前列对应的球是否已经被卡住，若已经被卡住直接跳过。若还未被卡住，判断球在当前位置是否会被卡住，若被卡住，标记当前球被卡住；若当前位置可以移动，标记位置为下一次移动出现的位置

代码：

public int[] findBall(int[][] grid) {
        int m = grid.length;
        int n = grid[0].length;
        int[] ball = new int[n];
        // 初始化所有球
        for (int i = 0; i < n; i++) {
            ball[i] = i;
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                // 若该列球已经被卡住，跳过
                if (ball[j] == -1) {
                    continue;
                }
                // 获取当前球所在位置
                int idx = ball[j];
                // 判断当前位置是否卡球
                if (idx == 0) {
                    // 判断若在左边界是否卡住
                    if (grid[i][idx] == -1 || (idx + 1 < n && grid[i][idx] == 1 && grid[i][idx + 1] == -1)) {
                        ball[j] = -1;
                    } else {
                        // 若未被卡住，还需要考虑列数只有 1 的情况
                        ball[j] = idx + 1 >= n ? -1 : idx + 1;
                    }
                } else if (idx == n - 1) {
                    // 判断若在右边界是否卡住
                    if (grid[i][idx] == 1 || (idx - 1 >= 0 && grid[i][idx] == -1 && grid[i][idx - 1] == 1)) {
                        ball[j] = -1;
                    } else {
                        // 若未被卡住，还需要考虑列数只有 1 的情况
                        ball[j] = idx - 1 < 0 ? -1 : idx - 1;
                    }
                } else {
                    // 判断是否与两侧网格卡住
                    if (grid[i][idx] == 1) {
                        ball[j] = grid[i][idx + 1] == 1 ? idx + 1 : -1;
                    } else {
                        ball[j] = grid[i][idx - 1] == -1 ? idx - 1 : -1;
                    }
                }
            }
        }
        return ball;
    }