地址：

5210. 球会落何处

状态：AC

代码：

  1. class Solution {
  2. public:
  3.     vector<int> findBall(vector<vector<int>>& grid) {
  4.         int m = grid.size();
  5.         int n = grid[0].size();
  6.         vector<int> ans(n,-1);
  7.         for(int i = 0;i<n;i++){
  8.             int temp = i;
  9.             for(int j = 0;j<m;j++){
  10.                 if(temp == 0 && grid[j][temp] == -1){
  11.                     temp = -1;
  12.                     break;
  13.                 }else if(temp == n-1 && grid[j][temp] == 1){
  14.                     temp = -1;
  15.                     break;
  16.                 }else{
  17.                     if((grid[j][temp] == grid[j][temp+grid[j][temp]])){
  18.                         temp += grid[j][temp];
  19.                     }else{
  20.                         temp = -1;
  21.                         break;
  22.                     }
  23.                 }
  24.             }
  25.             ans[i] = temp;
  26.         }
  27.         return ans;
  28.     }
  29. };