C++刷LeetCode(1706. 球会落何处)【深度优先搜索】

糖逗

题目描述：

1. 用一个大小为 m x n 的二维网格 grid 表示一个箱子。你有 n 颗球。箱子的顶部和底部都是开着的。
   
2. 
   
3. 箱子中的每个单元格都有一个对角线挡板，跨过单元格的两个角，可以将球导向左侧或者右侧。
   
4. 
   
5. 将球导向右侧的挡板跨过左上角和右下角，在网格中用 1 表示。
   
6. 将球导向左侧的挡板跨过右上角和左下角，在网格中用 -1 表示。
   
7. 在箱子每一列的顶端各放一颗球。每颗球都可能卡在箱子里或从底部掉出来。如果球恰好卡在两块挡板之间的 "V" 形图案，或者被一块挡导向到箱子的任意一侧边上，就会卡住。
   
8. 
   
9. 返回一个大小为 n 的数组 answer ，其中 answer[i] 是球放在顶部的第 i 列后从底部掉出来的那一列对应的下标，如果球卡在盒子里，则返回 -1 。
   
10. 
    
11.  
    
12. 
    
13. 示例 1：
    
14. 
    
15. 
    
16. 
    
17. 输入：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]]
    
18. 输出：[1,-1,-1,-1,-1]
    
19. 解释：示例如图：
    
20. b0 球开始放在第 0 列上，最终从箱子底部第 1 列掉出。
    
21. b1 球开始放在第 1 列上，会卡在第 2、3 列和第 1 行之间的 "V" 形里。
    
22. b2 球开始放在第 2 列上，会卡在第 2、3 列和第 0 行之间的 "V" 形里。
    
23. b3 球开始放在第 3 列上，会卡在第 2、3 列和第 0 行之间的 "V" 形里。
    
24. b4 球开始放在第 4 列上，会卡在第 2、3 列和第 1 行之间的 "V" 形里。
    
25. 示例 2：
    
26. 
    
27. 输入：grid = [[-1]]
    
28. 输出：[-1]
    
29. 解释：球被卡在箱子左侧边上。
    
30. 示例 3：
    
31. 
    
32. 输入：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]]
    
33. 输出：[0,1,2,3,4,-1]
    
34.  
    
35. 
    
36. 提示：
    
37. 
    
38. m == grid.length
    
39. n == grid[i].length
    
40. 1 <= m, n <= 100
    
41. grid[i][j] 为 1 或 -1
    
42. 
    
43. 来源：力扣（LeetCode）
    
44. 链接：https://leetcode-cn.com/problems/where-will-the-ball-fall
    
45. 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

复制代码

1. class Solution {
   
2. private:
   
3.     int len1, len2;
   
4.     vector<int>res;
   
5. public:
   
6.     int dfs(int x, int y, vector<vector<int> >&grid){
   
7.         if(x == len1)return y;
   
8.         if(grid[x][y] == 1){
   
9.             if(y+1 < len2 && grid[x][y+1] == 1){
   
10.                 return dfs(x+1, y+1, grid);
    
11.             }
    
12.         }else if(grid[x][y] == -1){
    
13.             if(y-1 >= 0 && grid[x][y-1] == -1){
    
14.                 return dfs(x+1, y-1, grid);
    
15.             }
    
16.         }
    
17.         return -1;
    
18.     }
    
19.     vector<int> findBall(vector<vector<int>>& grid) {
    
20.         len1 = grid.size(), len2 = grid[0].size();
    
21.         res = vector<int>(len2, -1);
    
22.         for(int i = 0 ; i < len2; i++){
    
23.             res[i] = dfs(0, i, grid);
    
24.         }
    
25.         return res;
    
26.     }
    
27. };

复制代码

参考链接：https://leetcode-cn.com/problems ... -by-yfxu_0209-n602/