Merge pull request soapyigu#283 from peacemoon/fix_keyboardrow

soapyigu · web-flow · commit 9251282a07c7 · 2020-01-20T16:10:24.000-08:00
Fix solution for KeyboardRow
diff --git a/String/KeyboardRow.swift b/String/KeyboardRow.swift
@@ -1,23 +1,23 @@
 /**
  * Question Link: https://leetcode.com/problems/keyboard-row/
  * Primary idea: Use filter to determine the word is subset or not.
- * 
+ *
  * Note: You can also use intersect() or union() functions to solve this problem.
  *
  * Time Complexity: O(nm), Space Complexity: O(n)
  *
  */
 
 class KeyboardRow {
-    func findWords(_ words: [String]) -> [String] {
-        let rowOne = "qwertyuiop", rowTwo = "asdfghjkl", rowThree = "zxcvbnm"
+  func findWords(_ words: [String]) -> [String] {
+      let rowOne = "qwertyuiop", rowTwo = "asdfghjkl", rowThree = "zxcvbnm"
 
-        return words.filter { word in rowOne.contains(word) || rowTwo.contains(word) || rowThree.contains(word) }
-    }
+      return words.filter { word in rowOne.contains(word.lowercased()) || rowTwo.contains(word.lowercased()) || rowThree.contains(word.lowercased()) }
+  }
+}
 
-    extension String {
-        func contains(_ word: String) -> Bool {
-            return word.filter { c in !self.contains(c) }.characters.count == 0
-        }
+extension String {
+    func contains(_ word: String) -> Bool {
+        return word.filter { c in !self.contains(c) }.count == 0
     }
-}
+}
