diff --git a/大厂刷题班/class02/Code04_Drive.java b/大厂刷题班/class02/Code04_Drive.java index a5a3a07..013fc30 100644 --- a/大厂刷题班/class02/Code04_Drive.java +++ b/大厂刷题班/class02/Code04_Drive.java @@ -81,6 +81,26 @@ public class Code04_Drive { return sum; } + // 找到了leetcode上的测试 + // leetcode上让求最小,课上讲的求最大 + // 其实是一个意思 + // 测试链接 : https://leetcode.cn/problems/two-city-scheduling/ + public static int twoCitySchedCost(int[][] costs) { + int N = costs.length; + int[] arr = new int[N]; + int sum = 0; + for (int i = 0; i < N; i++) { + arr[i] = costs[i][1] - costs[i][0]; + sum += costs[i][0]; + } + Arrays.sort(arr); + int M = N >> 1; + for (int i = 0; i < M; i++) { + sum += arr[i]; + } + return sum; + } + // 返回随机len*2大小的正数矩阵 // 值在0~value-1之间 public static int[][] randomMatrix(int len, int value) {