# Time:  O(1)
# Space: O(1)

class Solution(object):
    def flipLights(self, n, m):
        """
        :type n: int
        :type m: int
        :rtype: int
        """
        if m == 0:            return 1
        if n == 1:            return 2
        if m == 1 and n == 2: return 3
        if m == 1 or n == 2   return 4
        if m == 2:            return 7
        return 8

# Time:  O(1)
# Space: O(1)

class Solution(object):
    def flipLights(self, n, m):
        """
        :type n: int
        :type m: int
        :rtype: int
        """
        if m == 0:            return 1
        if n == 1:            return 2
        if m == 1 and n == 2: return 3
        if m == 1 or  n == 2: return 4
        if m == 2:            return 7
        return 8

Update bulb-switcher-ii.py