给定两个有序数组arr1和arr2，再给定一个正数K,求两个数累加和最大的前K个，两个数必须分别来自arr1和arr2

问题描述：

        给定两个有序数组arr1和arr2，再给定一个正数K,求两个数累加和最大的前K个，两个数必须分别来自arr1和arr2。

思路

        定义一个大根堆和一个二维数组，大根堆按照两个数的和大小进行排序，二维表行表示arr1列表示arr2其含义是str1和str2两下标的组合之前是否存在过。

代码

    //放入大根堆中的结构
    public static class Node {
        public int index1;//arr1中的位置
        public int index2;//arr2中的位置
        public int sum;// arr1[index1] + arr2[index2]的值

        public Node(int i1, int i2, int s) {
            index1 = i1;
            index2 = i2;
            sum = s;
        }
    }

    //生成大根堆的比较器
    public static class MaxHeapComp implements Comparator<Node> {

        @Override
        public int compare(Node o1, Node o2) {
            return o2.sum - o1.sum;
        }
    }

    public static int[] topKSum(int[] arr1, int[] arr2, int topK) {
        if (arr1 == null || arr2 == null || topK < 1) {
            return null;
        }
        //arr1 50   arr2 20   1000   topK 100万
        topK = Math.min(topK, arr1.length * arr2.length);

        int[] res = new int[topK];
        int resIndex = 0;
        PriorityQueue<Node> maxHeap = new PriorityQueue<>(new MaxHeapComp());

        //set[i][j] == false,arr1[i]  arr2[j]之前，没进过
        //set[i][j] == true,arr1[i]  arr2[j]之前，进过
        boolean[][] set = new boolean[arr1.length][arr2.length];
        int i1 = arr1.length - 1;
        int i2 = arr2.length - 1;
        maxHeap.add(new Node(i1, i2, arr1[i1] + arr2[i2]));
        set[i1][i2] = true;
        while (resIndex != topK) {
            Node curNode = maxHeap.poll();
            res[resIndex++] = curNode.sum;
            i1 = curNode.index1;
            i2 = curNode.index2;
            if (i1 - 1 >= 0 && !set[i1 - 1][i2]) {
                set[i1 - 1][i2] = true;
                maxHeap.add(new Node(i1 - 1, i2, arr1[i1 - 1] + arr2[i2]));
            }
            if (i2 - 1 >= 0 && !set[i1][i2 - 1]) {
                set[i1][i2 - 1] = true;
                maxHeap.add(new Node(i1, i2 - 1, arr1[i1] + arr2[i2 - 1]));
            }
        }
        return res;
    }