前言
对贪心的练习,是对逻辑分析考察的一个好方式。考察 脑容量/专注力/逻辑连续性/和编码融汇贯通程度(用编码体现逻辑)/优化方向的思考速度。
一、最小差值平方和
二、贪心 -> 贪心优化
1、基本贪心
贪心核心:选最大的gap来进行减一,让平方和下降的更快!
public long minSumSquareDiff(int[] nums1, int[] nums2, int k1, int k2) {
int n = nums1.length;
long[] gap = new long[n];
for (int i = 0; i < n; i++) gap[i] = Math.abs(nums1[i] - nums2[i]);
Arrays.sort(gap);
long sum = Arrays.stream(gap).sum();
long all = k1 + k2;
if (sum <= all) return 0;
while (all != 0) {
long begin = gap[gap.length - 1];
if (begin == 0) break;
for (int i = n - 1; i >= 0 && gap[i] == begin && all != 0; i--) {
gap[i]--;
all--;
}
}
long ans = 0;
for (long i : gap) ans += i * i;
return ans;
}
2、贪心优化
贪心核心:选最大的gap(相同的一组)来进行减一,让减法从O(n)到O(1),让平方和下降的更快!
// 改进版,一个个减太耗时,一组一起减一。
public long minSumSquareDiff2(int[] nums1, int[] nums2, int k1, int k2) {
int n = nums1.length;
Map<Long, Long> fx = new HashMap<>();
long max = 0;
for (int i = 0; i < n; i++) {
long gap = Math.abs(nums1[i] - nums2[i]);
max = Math.max(max, gap);
fx.put(gap, fx.getOrDefault(gap, 0l) + 1l);
}
int all = k1 + k2;
for (long i = max; i > 0 && all > 0; i--) {
if (!fx.containsKey(i)) continue;
if (all < fx.get(i)) {
fx.put(i - 1, fx.getOrDefault(i - 1, 0l) + all);
fx.put(i, fx.get(i) - all);
break;
}
fx.put(i - 1, fx.getOrDefault(i - 1, 0l) + fx.get(i));
all -= fx.get(i);
fx.remove(i);
}
long ans = 0;
Set<Map.Entry<Long, Long>> entries = fx.entrySet();
for (Map.Entry<Long, Long> entry : entries) {
long val = entry.getKey() * entry.getKey() * entry.getValue();
ans += val;
}
return ans;
}
总结
1)多参加比赛,用实际经历修正我的想法和计划,激发潜力,挑战极限。
2)多练习贪心算法,成长的高质量养料。
参考文献
[1] LeetCode 最小差值平方和