import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param pricesA int整型一维数组 * @param pricesB int整型一维数组 * @param k int整型 * @return int整型二维数组 */ public int[][] kPairsWithLargestSums (int[] pricesA, int[] pricesB, int k) { Map<int[], Integer> map = new HashMap<>(); for (int value : pricesA) { for (int i : pricesB) { map.put(new int[] {value, i}, value + i); } } List<Map.Entry<int[], Integer>> list = new ArrayList<>(map.entrySet()); list.sort((o1, o2) -> { if (o1.getValue() > o2.getValue()) { return -1; } else if (o1.getValue().equals(o2.getValue())) { return o2.getKey()[0] - o1.getKey()[0]; } else { return 1; } }); int[][] result = new int[k][2]; for (int i = 0; i < k; i++) { result[i] = list.get(i).getKey(); } return result; } }
本题知识点分析:
1.贪心算法
2.有序集合的存取
3.哈希表存取
4.集合自定义排序(Lambda表达式简写)
本题解题思路分析:
1.先将所有可能的总和存放到map集合中,以{1,2},3这样为例
2.利用List存取map集合并进行自定义排序,以value为降序,如果相等,以A组优先
3.取k个,就按顺序从集合中取