给你 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
class Solution:
def numberOfWeeks(self, milestones: List[int]) -> int:
mx, s = max(milestones), sum(milestones)
rest = s - mx
return rest * 2 + 1 if mx > rest + 1 else s
class Solution {
public long numberOfWeeks(int[] milestones) {
int mx = 0;
long s = 0;
for (int e : milestones) {
s += e;
mx = Math.max(mx, e);
}
long rest = s - mx;
return mx > rest + 1 ? rest * 2 + 1 : s;
}
}
class Solution {
public:
long long numberOfWeeks(vector<int>& milestones) {
int mx = *max_element(milestones.begin(), milestones.end());
long long s = accumulate(milestones.begin(), milestones.end(), 0LL);
long long rest = s - mx;
return mx > rest + 1 ? rest * 2 + 1 : s;
}
};
func numberOfWeeks(milestones []int) int64 {
mx, s := 0, 0
for _, e := range milestones {
mx = max(mx, e)
s += e
}
rest := s - mx
if mx > rest+1 {
return int64(rest*2 + 1)
}
return int64(s)
}
func max(a, b int) int {
if a > b {
return a
}
return b
}