500 - Python

Author: ashwek

solution/0500.Keyboard Row/Solution.py

@@ -0,0 +1,16 @@
+class Solution:
+    def findWords(self, words):
+        """
+        :type words: List[str]
+        :rtype: List[str]
+        """
+
+        Rows={"q":0, "w":0, "e":0, "r":0, "t":0, "y":0, "u":0, "i":0, "o":0, "p":0, "a":1, "s":1, "d":1, "f":1, "g":1, "h":1, "j":1, "k":1, "l":1, "z":2, "x":2, "c":2, "v":2, "b":2, "n":2, "m":2}
+
+        Final = []
+        for each in words :
+
+            if all([Rows[each[0].lower()]==Rows[ch] for ch in each.lower()]) :
+                Final.append(each)
+
+        return Final