500 Keyboard Row.py

#!/usr/bin/python3
"""
Given a List of words, return the words that can be typed using letters of
alphabet on only one row's of American keyboard like the image below.
Author: Rajeev Ranjan

"""


class Solution:
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        rows = [
            "qwertyuiop",
            "asdfghjkl",
            "zxcvbnm",
        ]
        d = {
            e: i
            for i, v in enumerate(rows)
            for e in v
        }
        return [
            w
            for w in words
            if all(d[w[0].lower()] == d[l.lower()] for l in w)
        ]


if __name__ == "__main__":
    assert Solution().findWords(["Hello", "Alaska", "Dad", "Peace"]) == ["Alaska", "Dad"]