### 题目描述 用一个大小为 m x n 的二维网格 grid 表示一个箱子。你有 n 颗球。箱子的顶部和底部都是开着的。 箱子中的每个单元格都有一个对角线挡板，跨过单元格的两个角，可以将球导向左侧或者右侧。 - 将球导向右侧的挡板跨过左上角和右下角，在网格中用 1 表示。 - 将球导向左侧的挡板跨过右上角和左下角，在网格中用 -1 表示。 在箱子每一列的顶端各放一颗球。每颗球都可能卡在箱子里或从底部掉出来。如果球恰好卡在两块挡板之间的 "V" 形图案，或者被一块挡导向到箱子的任意一侧边上，就会卡住。 返回一个大小为 n 的数组 answer ，其中 answer[i] 是球放在顶部的第 i 列后从底部掉出来的那一列对应的下标，如果球卡在盒子里，则返回 -1 。 ### 输入输出 #### 示例1  ``` 输入：grid = [[1,1,1,-1,-1],[1,1,1,-1,-1],[-1,-1,-1,1,1],[1,1,1,1,-1],[-1,-1,-1,-1,-1]] 输出：[1,-1,-1,-1,-1] 解释：示例如图： b0 球开始放在第 0 列上，最终从箱子底部第 1 列掉出。 b1 球开始放在第 1 列上，会卡在第 2、3 列和第 1 行之间的 "V" 形里。 b2 球开始放在第 2 列上，会卡在第 2、3 列和第 0 行之间的 "V" 形里。 b3 球开始放在第 3 列上，会卡在第 2、3 列和第 0 行之间的 "V" 形里。 b4 球开始放在第 4 列上，会卡在第 2、3 列和第 1 行之间的 "V" 形里。 ``` **提示** - m == grid.length - n == grid[i].length - 1 <= m, n <= 100 - grid[i][j] 为 1 或 -1 ### 题目解答 #### 思路 对于该问题采用模拟的方式，模拟每一行放一个球，每次使得球向下进行滚落，直到球落出或者球卡住，记录结果。 **算法复杂度** 时间复杂度：一共需要模拟n个球，每个球最多滚落m行，因此时间复杂度$$O(mn)$$ 空间复杂度：返回一个列表即可，中间变量的空间复杂度$$O(1)$$ #### 代码实现 ```java class Solution { public int[] findBall(int[][] grid) { int[] res = new int[grid[0].length]; // 每一列依次放一个球 for(int i=0;i<grid[0].length;i++){ // 一行一行判断 int row = 0; int col = i; int k = 0; for(k=0;k<grid.length;k++){ // 向右走的板子 if(grid[k][col] == 1){ // 如果是最后一列卡住了 if(col == grid[0].length-1){ res[i] = -1; break; } // 不是最后一列 else{ // 右边一个板子也向右 if(grid[k][col+1]==1) col++; else{ res[i] = -1; break; } } }else if(grid[k][col] == -1){ // 如果是第一列卡住了 if(col == 0){ res[i] = -1; break; } // 不是第一列 else{ // 左边一个板子也向左 if(grid[k][col-1]==-1) col--; else{ res[i] = -1; break; } } } } // 最后漏出 if(k == grid.length){ res[i] = col; } } return res; } } ``` 最后编辑：2024年04月23日 ©著作权归作者所有 赞 0 分享
最新回复