Create 1706.Where-Will-the-Ball-Fall.cpp

yanxinpan · Dec 28, 2020 · 3aeafb0 · 3aeafb0
1 parent 8fb6140
commit 3aeafb0
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/Others/1706.Where-Will-the-Ball-Fall/1706.Where-Will-the-Ball-Fall.cpp b/Others/1706.Where-Will-the-Ball-Fall/1706.Where-Will-the-Ball-Fall.cpp
@@ -0,0 +1,38 @@
+class Solution {
+public:
+    vector<int> findBall(vector<vector<int>>& grid) 
+    {
+        int m = grid.size();
+        int n = grid[0].size();
+
+        vector<int>ret(n,-1);
+        for (int j=0; j<n; j++)
+        {
+            int x = 0, y = j;
+            while (x < m)
+            {
+                if (grid[x][y]==1 && y==n-1)
+                    break;
+                else if (grid[x][y]==-1 && y==0)
+                    break;
+                else if (grid[x][y]==1 && grid[x][y+1]==-1)
+                    break;
+                else if (grid[x][y]==-1 && grid[x][y-1]==1)
+                    break;
+                else if (grid[x][y]==1)
+                {
+                    x++;
+                    y++;
+                }
+                else
+                {
+                    x++;
+                    y--;
+                }
+            }
+            if (x==m)
+                ret[j] = y;
+        }
+        return ret;
+    }
+};