Easy
Categorize Box According to Criteria — Python
Full explanation · Time O(1) · Space O(1)
# Time: O(1)
# Space: O(1)
# math, implementation
class Solution(object):
def categorizeBox(self, length, width, height, mass):
"""
:type length: int
:type width: int
:type height: int
:type mass: int
:rtype: str
"""
bulky = any(x >= 10**4 for x in (length, width, height)) or length*width*height >= 10**9
heavy = mass >= 100
if bulky and heavy:
return "Both"
if bulky:
return "Bulky"
if heavy:
return "Heavy"
return "Neither"
# Time: O(1)
# Space: O(1)
# math, implementation
class Solution2(object):
def categorizeBox(self, length, width, height, mass):
"""
:type length: int
:type width: int
:type height: int
:type mass: int
:rtype: str
"""
CATEGORIES = ["Neither", "Heavy", "Bulky", "Both"]
i = 2*(any(x >= 10**4 for x in (length, width, height)) or length*width*height >= 10**9)+int(mass >= 100)
return CATEGORIES[i]