[leetcode] 球会落何处 模拟
给出一个矩阵,里面的值为-1 or 1
-1的时候是从左上到右下,1的时候是从左下到右上
当一个球从上方x(0 < x < m)放入之后,从哪个出口掉落还是无法从出口掉落
能通过的情况为:
- / / 即某条线为’/’,其左边的线也是’/’,箭头所指为当前斜线
{\ } ↑ - \ \ 即当前线为’’,其右边的线也是’’,箭头所指为当前斜线
{} ↑
但是还要注意边界问题
Code:
class Solution {
public:
vector<int> findBall(vector<vector<int>>& grid) {
int n = grid.size(), m = grid[0].size();
vector<int> ans;
for(int i = 0; i < m;i ++) {
int x = 0,y = i;
while(x < n && y >= 0 && y < m) {
if(y >= 1 && grid[x][y] == grid[x][y-1] && grid[x][y] == -1) { // '/'
++ x;
-- y;
}else if(y < m - 1 && grid[x][y] == grid[x][y+1] && grid[x][y] == 1) { // '\'
++ x;
++ y;
}
else break;
}
if(x == n) ans.push_back(y);
else ans.push_back(-1);
}
return ans;
}
};