LeetCode：1953. 你可以工作的最大周数（贪心 Java）

目录

1953. 你可以工作的最大周数

题目描述：

实现代码与解析：

贪心：

原理思路：

---

1953. 你可以工作的最大周数

题目描述：

        给你 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 {
    public static void main(String[] args) {
        int[] q = new int[]{1,2,3};
        long l = numberOfWeeks(q);
        System.out.println(l);
    }
    public static long numberOfWeeks(int[] milestones) {

        PriorityQueue<int[]> pq = new PriorityQueue<>((a, b) -> b[0] - a[0]); // 大顶堆
        int n = milestones.length;
        for (int i = 0; i < n; i++) {
            pq.offer(new int[] {milestones[i], i});
        }

        int res = 0;
        int pre = -1;
        List<int[]> list = new ArrayList<>();
        while (true) {
            if (pq.size() == 0) break;

            int[] t = pq.peek();
            pq.poll();
            if (t[1] != pre) {
                res++;
                t[0]--;
                if (t[0] != 0) {
                    pq.offer(t);
                }
                for (int i = 0; i < list.size(); i++) {
                    pq.offer(list.get(i));
                }
                pre = t[1];
                list.clear();
            } else {
                list.add(t);
            }
        }
        return res;
    }
}

这种方法不行的话，说明还有更加简单的方法，比如数学公式。果然这题是一个公式题。

class Solution {
    public long numberOfWeeks(int[] milestones) {
        int mx = 0;
        long sum = 0;
        for (int t : milestones) {
            sum += t;
            mx = Math.max(mx, t);
        }
        long diff = sum - mx;
        return mx > diff + 1 ? diff * 2 + 1 : sum;
    }
}

原理思路：

        额，可以理解为小学种树问题，肯定是有先种数量最多的那种树，但是不能连续，所以中间穿插其他树，这么就可以得出，只要其他树的个数 + 1小于最多的那种树的数量，就可以完成所有任务，否则就是其余树的个数 * 2 + 1。把这个看成小学种树问题就可以很好理解为什么了。