sy1214ei 님의 블로그

[Leet Code] 500. Keyboard Row - Python 본문

[Coding]

[Leet Code] 500. Keyboard Row - Python

sy1214ei 2024. 12. 23. 16:06

https://leetcode.com/problems/keyboard-row/description/

level : easy
class Solution:
    def findWords(self, words: List[str]) -> List[str]:
        set1 = {'q','w','e','r','t','y','u','i','o','p'}
        set2 = {'a','s','d','f','g','h','j','k','l'}
        set3 = {'z','x','c','v','b','n','m'}
        
        res = []
        for i in words:
            wordset = set(i.lower())
            if (wordset&set1 == wordset) or (wordset&set2 == wordset) or (wordset&set3 == wordset):
                res.append(i)
        return res
        
        # Time Complexity : O(n)
        # Space Complexity : O(n)