1706.WhereWillTheBallFall.cpp

#include<bits/stdc++.h>
using namespace std;

class Solution {
public:
    vector<int> findBall(vector<vector<int>>& grid) {
        
        
        vector<int>res;
        
        int m=grid.size(),n=grid[0].size();
        
        for(int i=0;i<n;i++)
        {
            
            int i1=i,i2;
            
            for(int j=0;j<m;j++)
            {
                i2=i1+grid[j][i1];
                
                if(i2<0 || i2>=n || grid[j][i2]!=grid[j][i1])
                {
                    i1=-1;
                    break;
                }
                 i1=i2;
                 
            }
            res.push_back(i1);
            
        }
        
        return res;
        
    }
};