LeetCode——2335. 装满杯子需要的最短总时长

一、题目

现有一台饮水机，可以制备冷水、温水和热水。每秒钟，可以装满 2 杯 不同 类型的水或者 1 杯任意类型的水。

给你一个下标从 0 开始、长度为 3 的整数数组 amount ，其中 amount[0]、amount[1] 和 amount[2] 分别表示需要装满冷水、温水和热水的杯子数量。返回装满所有杯子所需的 最少 秒数。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/minimum-amount-of-time-to-fill-cups/solutions/

二、C++解法

我的思路及代码

先对数组进行排序，然后每次从最多的两个中取 1 个然后 ans 加 1 ，直到其中一个为空，此时第二多的一定还剩 1 个，返回ans+最多的剩下的数字+1就是答案。

class Solution {
public:
    int fillCups(vector<int>& amount) {
        int ans=0;
        sort(amount.begin(),amount.end());
        while(amount[2]){
            ans++;
            amount[2] --;
            if(amount[0]){
                amount[1] --;
                sort(amount.begin(),amount.end());
            }else{ //表明已经取空了一个
                ans+=amount[2];
                break;
            }
        }
        return ans;
    }
};

• 时间复杂度：O(n)，n为要接的数量
• 空间复杂度：O(1)

官方参考代码

官方很巧妙的运用了数学的办法

class Solution {
public:
    int fillCups(vector<int>& amount) {
        sort(amount.begin(), amount.end());
        if (amount[2] > amount[1] + amount[0]) {
            return amount[2];
        }
        return (accumulate(amount.begin(), amount.end(), 0) + 1) / 2;
    }
};

但是我觉得他写复杂了，强行一行代码搞定

class Solution {
public:
    int fillCups(vector<int>& amount) {
        return  sort(amount.begin(),amount.end()),amount[0]+amount[1]<amount[2]?amount[2]:(amount[0]+amount[1]+amount[2]+1)/2;
    }
};

• 时间复杂度：O(1)
• 空间复杂度：O(1)