algorithm-class/算法周更班/class_2022_06_1_week/Code01_WhereWillTheBallFall.java

package class_2022_06_1_week;

// 最好打开链接看图
// 用一个大小为 m x n 的二维网格 grid 表示一个箱子
// 你有 n 颗球。箱子的顶部和底部都是开着的。
// 箱子中的每个单元格都有一个对角线挡板，跨过单元格的两个角，
// 可以将球导向左侧或者右侧。
// 将球导向右侧的挡板跨过左上角和右下角，在网格中用 1 表示。
// 将球导向左侧的挡板跨过右上角和左下角，在网格中用 -1 表示。
// 在箱子每一列的顶端各放一颗球。每颗球都可能卡在箱子里或从底部掉出来。
// 如果球恰好卡在两块挡板之间的 "V" 形图案，或者被一块挡导向到箱子的任意一侧边上，就会卡住。
// 返回一个大小为 n 的数组 answer ，
// 其中 answer[i] 是球放在顶部的第 i 列后从底部掉出来的那一列对应的下标，
// 如果球卡在盒子里，则返回 -1
// 本题测试链接 : https://leetcode.com/problems/where-will-the-ball-fall/
public class Code01_WhereWillTheBallFall {

	public static int[] findBall(int[][] grid) {
		int n = grid.length;
		int m = grid[0].length;
		int[] ans = new int[m];
		for (int col = 0; col < m; col++) {
			// (0,0)  (0,1)  (0,2)
			int i = 0;
			int j = col;
			while (i < n) {
				// (i,j)  左上 -> 右下的格子  grid[i][j] == 1
				// (i+1, j+1)
				// (i,j)  右上 -> 左下的格子  grid[i][j] == -1
				// (i+1, j-1)
				int jnext = j + grid[i][j];
				if (jnext < 0 || jnext == m || grid[i][j] != grid[i][jnext]) {
					ans[col] = -1;
					break;
				}
				i++;
				j = jnext;
			}
			if (i == n) {
				ans[col] = j;
			}
		}
		return ans;
	}

}