LeetCode 筆記 - 1710. Maximum Units on a Truck

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        # truck size: the number of boxes that can be put on the truck
        
        result = 0
        
        boxTypes.sort(key=lambda x: -x[1])
        
        result = 0
        for boxType in boxTypes:
            
            boxes = (truckSize if truckSize <= boxType[0] else boxType[0])
            
            truckSize -= boxes
            result += boxes * boxType[1]
            
            if truckSize == 0:
                break
        
        return result