-
Notifications
You must be signed in to change notification settings - Fork 0
/
1706.cpp
45 lines (41 loc) · 1.17 KB
/
1706.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
vector<int> findBall(vector<vector<int>>& grid) {
int m=grid.size(); int n=grid[0].size();
vector<int>results;
for(int i=0;i<n;i++){
int x=0; int y=i;
while(x<m){
if((y==0 and grid[x][y]==-1) or (y==n-1 and grid[x][y]==1)){
results.push_back(-1);
break;
}
if(grid[x][y]==1){
if(grid[x][y+1]==-1){
results.push_back(-1);
break;
}else{
x+=1; y+=1;
}
}else if(grid[x][y]==-1){
if(grid[x][y-1]==1){
results.push_back(-1);
break;
}else{
x+=1; y-=1;
}
}
}
if(x==m){
results.push_back(y);
}
}
return results;
}
};
int main(){
return 0;
}