Leetcode 1710. Maximum Units on a Truck

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        count = 0
        boxTypes.sort(key=lambda x: x[1], reverse=True)
        i = 0
        while truckSize > 0 and i < len(boxTypes):
            if truckSize - boxTypes[i][0] > 0:
                truckSize -= boxTypes[i][0]
                count += boxTypes[i][0] * boxTypes[i][1]
            else:
                count += truckSize * boxTypes[i][1]
                truckSize = 0
            i += 1
        return count