1402. Reducing Dishes

A chef has collected data on the satisfaction level of his n dishes. Chef can cook any dish in 1 unit of time.

Like-time coefficient of a dish is defined as the time taken to cook that dish including previous dishes multiplied by its satisfaction level i.e. time[i]*satisfaction[i]

Return the maximum sum of Like-time coefficient that the chef can obtain after dishes preparation.

Dishes can be prepared in any order and the chef can discard some dishes to get this maximum value.

class Solution {
public:
    int maxSatisfaction(vector<int>& satisfaction) {
        sort(satisfaction.begin(), satisfaction.end());
        if(satisfaction.back() <= 0) return 0;
        vector<int> sum{0};
        int res = 0;
        for(auto& n : satisfaction) {
            sum.push_back(sum.back() + n);
            res += n * (sum.size() - 1);
        }
        for(int i = 0; i < sum.size(); i++) {
            if(sum.back() - sum[i] >= 0) break;
            res -= (sum.back() - sum[i]);
        }
        return res;
    }
 };