LeetCode——堆

最新推荐文章于 2023-08-20 14:51:00 发布

K. Bob

最新推荐文章于 2023-08-20 14:51:00 发布

阅读量249

点赞数 2

分类专栏：算法

本文链接：https://blog.csdn.net/ThreeAspects/article/details/104461295

版权

算法专栏收录该内容

28 篇文章 2 订阅

订阅专栏

接雨水

42.接雨水（Stack）
407.接雨水II（PriorityQueue）

Kth问题

703.数据流中的第K大元素
 378.有序矩阵中第K小的元素
 215.数组中的第K个最大元素
 692.前K个高频单词
 347.前K个高频元素
给定两个有序数组arr1和arr2，再给定一个整数k，返回来自arr1和arr2的两个数相加和最大的前k个，两个数必须分别来自两个数组。时间复杂度: O(klogk); 辅助空间复杂度: O(k)

public class Solution {
    public class Node {
        public int index1;
        public int index2;
        public int value;
        public Node(int i1, int i2, int sum) {
            index1 = i1;
            index2 = i2;
            value = sum;
        }
    }
    // 大顶堆比较器
    public class MaxHeapComp implements Comparator<Node> {
        @Override
        public int compare(Node o1, Node o2) {
            return o2.value - o1.value;
        }
    }
    public int[] topKSum(int[] arr1, int[] arr2, int topK) {
        if (arr1 == null || arr2 == null || topK < 1) {
            return null;
        }
        topK = Math.min(topK, arr1.length * arr2.length);
        int[] res = new int[topK];
        int resIndex = 0;
        PriorityQueue<Node> maxHeap = new PriorityQueue<>(new MaxHeapComp());
        HashSet<String> positionSet = new HashSet<String>();
        int i1 = arr1.length - 1;
        int i2 = arr2.length - 1;
        maxHeap.add(new Node(i1, i2, arr1[i1] + arr2[i2]));
        positionSet.add(String.valueOf(i1 + "_" + i2));
        while (resIndex != topK) {
            // 弹出堆顶
            Node curNode = maxHeap.poll();
            res[resIndex++] = curNode.value;
            i1 = curNode.index1;
            i2 = curNode.index2;
            if (!positionSet.contains(String.valueOf(i1 - 1 + "_" + i2))) {
                positionSet.add(String.valueOf(i1 - 1, i2, arr1[i1 - 1] + arr2[i2]));
                maxHeap.add(new Node(i1 - 1, i2, arr1[i1 - 1] + arr2[i2]));
            }
            if (!positionSet.contains(String.valueOf(i1 + "_" + i2 - 1))) {
                positionSet.add(String.valueOf(i1 + "_" + i2 - 1));
                maxHeap.add(new Node(i1 , i2 - 1, arr1[i1] + arr2[i2 -1]));
            }
        }
        return res;
    }
}