目录
一、题目
1、题目描述
给你
n
个项目,编号从0
到n - 1
。同时给你一个整数数组milestones
,其中每个milestones[i]
表示第i
个项目中的阶段任务数量。你可以按下面两个规则参与项目中的工作:
- 每周,你将会完成 某一个 项目中的 恰好一个 阶段任务。你每周都 必须 工作。
- 在 连续的 两周中,你 不能 参与并完成同一个项目中的两个阶段任务。
一旦所有项目中的全部阶段任务都完成,或者仅剩余一个阶段任务都会导致你违反上面的规则,那么你将 停止工作 。注意,由于这些条件的限制,你可能无法完成所有阶段任务。
返回在不违反上面规则的情况下你 最多 能工作多少周。
2、接口描述
python3
class Solution:
def numberOfWeeks(self, milestones: List[int]) -> int:
cpp
class Solution {
public:
long long numberOfWeeks(vector<int>& milestones) {
}
};
3、原题链接
二、解题报告
1、思路分析
对于给定的任务数组我们可以像上图那样从小到大进行排列,那么我们把每一行拿出来拼接就能得到一个合法序列,最后一列多出来的部分我们进行插入即可
下面分情况讨论:
记最大值为ma,前n - 1个任务和为s(注意不包括ma),第i个任务数目为c[i]
s - 1 = ma, s = ma, s + 1= ma时,我们总能将ma - c[n - 1]个任务插入到每一排拼接得到的合法序列中去,此时的合法排列长度为s+ma
ma < s - 1时,由于ma > c[n - 1],那么ma总可以使得前n - 1个任务都合法,然后由于s - 1 > ma,故剩余位置足够最后一列的剩余任务插入,此时的合法排列长度为s+ma
ma > s - 1时,我们在满足前n - 1个任务后,剩余位置不够最后一列剩余任务插入,故此时最大合法排列长度为s * 2 + 1
故,ma > s + 1时,答案为s * 2 + 1
ma <= s + 1时,答案为s + ma
我们发现ma > s + 1 等价位 s + ma > s * 2 + 1,故进一步简化答案为min(s * 2 + 1, s + ma)
2、复杂度
时间复杂度: O(n)空间复杂度:O(1)
3、代码详解
python3
class Solution:
def numberOfWeeks(self, milestones: List[int]) -> int:
s = sum(milestones)
ma = max(milestones)
s -= ma
return min(s * 2 + 1, ma + s)
cpp
class Solution {
public:
typedef long long LL;
long long numberOfWeeks(vector<int>& milestones) {
LL s = accumulate(milestones.begin(), milestones.end(), 0LL);
int ma = *max_element(milestones.begin(), milestones.end());
return min(s, (s - ma) * 2 + 1);
}
};