Update Valid Word Square.java

varunu28 · web-flow · commit 2446f4b5c48d · 2022-12-07T13:32:53.000-08:00
diff --git a/Easy/Valid Word Square.java b/Easy/Valid Word Square.java
@@ -1,16 +1,13 @@
 class Solution {
-  public boolean validWordSquare(List<String> words) {
-    return IntStream.range(0, words.size())
-        .allMatch(i -> words.get(i).equals(verticalRepresentation(words, i)));
-  }
-
-  private String verticalRepresentation(List<String> words, int col){
-    StringBuilder sb = new StringBuilder();
-    for (String word : words) {
-      if (col < word.length()) {
-        sb.append(word.charAt(col));
-      }
+    public boolean validWordSquare(List<String> words) {
+        int n = words.size();
+        for (int i = 0; i < n; i++){
+            for (int j = 0; j < words.get(i).length(); j++){
+                if(j >= n || words.get(j).length() <= i || words.get(j).charAt(i) != words.get(i).charAt(j)) {
+                    return false;
+                }
+            }
+        }
+        return true;
     }
-    return sb.toString();
-  }
 }
