Update Battleships in a board.java

varunu28 · web-flow · commit ede04e0150f0 · 2022-09-20T13:05:14.000-07:00
diff --git a/Medium/Battleships in a board.java b/Medium/Battleships in a board.java
@@ -1,20 +1,17 @@
 class Solution {
-    public int countBattleships(char[][] board) {
-        int n = board.length;
-        if (n == 0) return 0;
-        int m = board[0].length;
-        
-        int count = 0;
-        for (int i=0;i<n;i++) {
-            for (int j=0;j<m;j++) {
-                if (board[i][j] == '.') continue;
-                if (i > 0 && board[i-1][j] == 'X')continue;
-                if (j > 0 && board[i][j-1] == 'X')continue;
-                
-                count++;
-            }
+  
+  public int countBattleships(char[][] board) {
+    int count = 0;
+    int rows = board.length;
+    int cols = board[0].length;
+    for (int i = 0; i < rows; i++) {
+      for (int j = 0; j < cols; j++) {
+        if (board[i][j] == '.' || (i > 0 && board[i - 1][j] == 'X') || (j > 0 && board[i][j - 1] == 'X')) {
+          continue;
         }
-        
-        return count;
+        count++;
+      }
     }
+    return count;
+  }
 }
