[412. Fizz Buzz] #leetcode

412_1007.java

class Solution(object):
// O(n) O(n)
    def fizzBuzz(self, n):
        """
        :type n: int
        :rtype: List[str]
        """
        res = []
        
        for i in range(1, n+1):
            three = i % 3 == 0
            five = i % 5 == 0
            if three and five:
                res.append("FizzBuzz")
            elif three:
                res.append("Fizz")
            elif five:
                res.append("Buzz")
            else:
                res.append(str(i))
        
        return res
        
class Solution(object):
    def fizzBuzz(self, n):
        """
        :type n: int
        :rtype: List[str]
        """
        return ["Fizz"*(i%3==0) + "Buzz"*(i%5==0) or str(i) for i in range(1, n+1)]