modify code

Author: algorithmzuo

ppt/算法讲解092【必备】贪心经典题目专题4.pptx

@@ -17,9 +17,14 @@
 public class Code01_MinimizeDeviation {
 
 	public static int minimumDeviation(int[] nums) {
+		// 有序表可以查询最大值、最小值
 		TreeSet<Integer> set = new TreeSet<>();
 		for (int num : nums) {
-			set.add(num % 2 == 0 ? num : num * 2);
+			if (num % 2 == 0) {
+				set.add(num);
+			} else {
+				set.add(num * 2);
+			}
 		}
 		int ans = set.last() - set.first();
 		while (ans > 0 && set.last() % 2 == 0) {

src/class092/Code01_MinimizeDeviation.java

@@ -4,28 +4,29 @@
 
 // 森林中的兔子
 // 森林中有未知数量的兔子
-// 提问其中若干只兔子 "还有多少只兔子与你颜色相同?"
-// 将答案收集到一个整数数组answers中
-// 其中answers[i]是第i只兔子的回答
-// 每只兔子都不会说谎，返回森林中兔子的最少数量
+// 你问兔子们一个问题: "还有多少只兔子与你颜色相同?"
+// 你将答案收集到了一个数组answers中
+// 你可能没有收集到所有兔子的回答，可能只是一部分兔子的回答
+// 其中answers[i]是第i只兔子的答案
+// 所有兔子都不会说错，返回森林中兔子的最少数量
 // 测试链接 : https://leetcode.cn/problems/rabbits-in-forest/
 public class Code02_RabbitsInForest {
 
 	public static int numRabbits(int[] arr) {
+		// a / b 向上取整 -> (a + b - 1) / b
 		Arrays.sort(arr);
-		int x = arr[0];
-		int cnt = 1;
+		int n = arr.length;
 		int ans = 0;
-		for (int i = 1; i < arr.length; i++) {
-			if (x != arr[i]) {
-				ans += ((cnt + x) / (x + 1)) * (x + 1);
-				x = arr[i];
-				cnt = 1;
-			} else {
-				cnt++;
+		for (int i = 0, j = 1, x; i < n; j++) {
+			x = arr[i];
+			while (j < n && x == arr[j]) {
+				j++;
 			}
+			// i...j-1 都是同一种答案，当前组有j-i个回答
+			ans += (j - i + x) / (x + 1) * (x + 1);
+			i = j;
 		}
-		return ans + ((cnt + x) / (x + 1)) * (x + 1);
+		return ans;
 	}
 
 }

src/class092/Code02_RabbitsInForest.java

@@ -10,14 +10,14 @@
 // 令 nums[j] = nums[j] - 2
 // 如果两个数组中每个元素出现的频率相等，我们称两个数组是 相似 的
 // 请你返回将 nums 变得与 target 相似的最少操作次数
-// 测试数据保证 nums 一定能变得与 target 相似
+// 测试数据保证nums一定能变得与target相似
 // 测试链接 : https://leetcode.cn/problems/minimum-number-of-operations-to-make-arrays-similar/
 public class Code03_MinimumOperationsMakeSimilar {
 
 	public static long makeSimilar(int[] nums, int[] target) {
 		int n = nums.length;
-		int oddSize = split(nums);
-		split(target);
+		int oddSize = split(nums, n);
+		split(target, n);
 		Arrays.sort(nums, 0, oddSize);
 		Arrays.sort(nums, oddSize, n);
 		Arrays.sort(target, 0, oddSize);
@@ -26,17 +26,19 @@ public static long makeSimilar(int[] nums, int[] target) {
 		for (int i = 0; i < n; i++) {
 			ans += Math.abs((long) nums[i] - target[i]);
 		}
-		return ans >> 2;
+		return ans / 4;
 	}
 
-	public static int split(int[] arr) {
-		int line = 0;
-		for (int i = 0; i < arr.length; i++) {
-			if ((arr[i] & 1) != 0) {
-				swap(arr, i, line++);
+	// 把数组分割成左部分全是奇数，右部分全是偶数
+	// 返回左部分的长度
+	public static int split(int[] arr, int n) {
+		int oddSize = 0;
+		for (int i = 0; i < n; i++) {
+			if ((arr[i] & 1) == 1) {
+				swap(arr, i, oddSize++);
 			}
 		}
-		return line;
+		return oddSize;
 	}
 
 	public static void swap(int[] arr, int i, int j) {

src/class092/Code03_MinimumOperationsMakeSimilar.java

@@ -51,8 +51,8 @@ public static void main(String[] args) throws IOException {
 
 	public static int compute() {
 		Arrays.sort(nums, 0, n, (a, b) -> Math.abs(a[0] - a[1]) - Math.abs(b[0] - b[1]));
-		int maxA = nums[0][0];
-		int maxB = nums[0][1];
+		int maxA = nums[0][0]; // 左边最大的推理能力
+		int maxB = nums[0][1]; // 左边最大的阅读能力
 		int ans = 0;
 		for (int i = 1; i < n; i++) {
 			if (nums[i][0] <= nums[i][1]) {

src/class092/Code04_Quiz.java

@@ -1,14 +1,17 @@
 package class092;
 
 // 将数组分成几个递增序列
-// 给你一个 非递减 的正整数数组 nums 和整数 K
+// 给你一个有序的正数数组 nums 和整数 K
 // 判断该数组是否可以被分成一个或几个 长度至少 为 K 的 不相交的递增子序列
+// 数组中的所有数字，都要被，若干不相交的递增子序列包含
 // 测试链接 : https://leetcode.cn/problems/divide-array-into-increasing-sequences/
 public class Code05_DivideArrayIncreasingSequences {
 
 	public static boolean canDivideIntoSubsequences(int[] nums, int k) {
 		int cnt = 1;
+		// maxCnt : 最大词频
 		int maxCnt = 1;
+		// 在有序数组中，求某个数的最大词频
 		for (int i = 1; i < nums.length; i++) {
 			if (nums[i - 1] != nums[i]) {
 				maxCnt = Math.max(maxCnt, cnt);
@@ -18,6 +21,8 @@ public static boolean canDivideIntoSubsequences(int[] nums, int k) {
 			}
 		}
 		maxCnt = Math.max(maxCnt, cnt);
+		// 向下取整如果满足 >= k
+		// 那么所有的递增子序列长度一定 >= k
 		return nums.length / maxCnt >= k;
 	}
 

src/class092/Code05_DivideArrayIncreasingSequences.java

@@ -20,7 +20,9 @@ public static int minRefuelStops(int target, int startFuel, int[][] stations) {
 		if (startFuel >= target) {
 			return 0;
 		}
+		// 大根堆
 		PriorityQueue<Integer> heap = new PriorityQueue<>((a, b) -> b - a);
+		// 包括初始油量 + 沿途加的油，能让你达到什么位置，to
 		int to = startFuel;
 		int cnt = 0;
 		for (int[] station : stations) {
@@ -40,6 +42,8 @@ public static int minRefuelStops(int target, int startFuel, int[][] stations) {
 			}
 			heap.add(fuel);
 		}
+		// 代码能走到这里，说明还没到达target
+		// 如果还有油，看看能不能冲到target
 		while (!heap.isEmpty()) {
 			to += heap.poll();
 			cnt++;