最小差值平方和[贪心练习 -＞基本贪心到优化贪心]

最新推荐文章于 2022-08-17 16:05:04 发布

REN_林森

最新推荐文章于 2022-08-17 16:05:04 发布

阅读量392

点赞数 1

分类专栏：数据机构与算法文章标签：贪心算法贪心优化 Java

本文链接：https://blog.csdn.net/qq_43164662/article/details/125712548

版权

数据机构与算法专栏收录该内容

399 篇文章 2 订阅

订阅专栏

贪心练习 -> 基本贪心到优化贪心

前言
一、最小差值平方和
二、贪心 -> 贪心优化
- 1、基本贪心
- 2、贪心优化
总结
参考文献

前言

对贪心的练习，是对逻辑分析考察的一个好方式。考察脑容量/专注力/逻辑连续性/和编码融汇贯通程度(用编码体现逻辑)/优化方向的思考速度。

一、最小差值平方和

在这里插入图片描述

二、贪心 -> 贪心优化

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;
    }