leetcode 5831. 你可以工作的最大周数（C++、java、python）

给你 n 个项目，编号从 0 到 n - 1 。同时给你一个整数数组 milestones ，其中每个 milestones[i] 表示第 i 个项目中的阶段任务数量。

你可以按下面两个规则参与项目中的工作：

• 每周，你将会完成 某一个 项目中的 恰好一个 阶段任务。你每周都 必须 工作。
• 在 连续的 两周中，你 不能 参与并完成同一个项目中的两个阶段任务。

一旦所有项目中的全部阶段任务都完成，或者仅剩余一个阶段任务都会导致你违反上面的规则，那么你将 停止工作 。注意，由于这些条件的限制，你可能无法完成所有阶段任务。

返回在不违反上面规则的情况下你 最多 能工作多少周。

示例 1：

输入：milestones = [1,2,3]
输出：6
解释：一种可能的情形是：
​​​​- 第 1 周，你参与并完成项目 0 中的一个阶段任务。
- 第 2 周，你参与并完成项目 2 中的一个阶段任务。
- 第 3 周，你参与并完成项目 1 中的一个阶段任务。
- 第 4 周，你参与并完成项目 2 中的一个阶段任务。
- 第 5 周，你参与并完成项目 1 中的一个阶段任务。
- 第 6 周，你参与并完成项目 2 中的一个阶段任务。
总周数是 6 。

示例 2：

输入：milestones = [5,2,1]
输出：7
解释：一种可能的情形是：
- 第 1 周，你参与并完成项目 0 中的一个阶段任务。
- 第 2 周，你参与并完成项目 1 中的一个阶段任务。
- 第 3 周，你参与并完成项目 0 中的一个阶段任务。
- 第 4 周，你参与并完成项目 1 中的一个阶段任务。
- 第 5 周，你参与并完成项目 0 中的一个阶段任务。
- 第 6 周，你参与并完成项目 2 中的一个阶段任务。
- 第 7 周，你参与并完成项目 0 中的一个阶段任务。
总周数是 7 。
注意，你不能在第 8 周参与完成项目 0 中的最后一个阶段任务，因为这会违反规则。
因此，项目 0 中会有一个阶段任务维持未完成状态。

提示：

• n == milestones.length
• 1 <= n <= 105
• 1 <= milestones[i] <= 109

C++

class Solution {
public:
    long long numberOfWeeks(vector<int>& milestones) {
        long long ans=0;
        long long sum=0;
        for(int i=0;i<milestones.size();i++) {
            sum+=milestones[i];
        }
        int max_val=*max_element(milestones.begin(),milestones.end());
        if(max_val*2<=sum+1) {
            return sum;
        }
        return (sum-max_val)*2+1;
    }
};

java

class Solution {
    public long numberOfWeeks(int[] milestones) {
        int n = milestones.length;
        int max_val = 0;
        long sum_val = 0;
        for (int i = 0; i < n; i++) {
            max_val = Math.max(max_val, milestones[i]);
            sum_val += milestones[i];
        }
        if (max_val * 2 <= sum_val + 1) {
            return sum_val;
        }
        return (sum_val - max_val) * 2 + 1;
    }
}

python

class Solution:
    def numberOfWeeks(self, milestones: List[int]) -> int:
        max_val = max(milestones)
        sum_val = sum(milestones)
        if max_val * 2 <= sum_val + 1:
            return sum_val
        return (sum_val - max_val) * 2 + 1