373. Find K Pairs with Smallest Sums

最新推荐文章于 2023-11-07 08:50:13 发布

原创最新推荐文章于 2023-11-07 08:50:13 发布 · 178 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#函数 #java #class

leetcode刷题专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种使用Java编程语言的方法，该方法通过计算两个整数数组的所有可能组合，然后找出这些组合中和最小的k对。具体实现包括创建组合、排序以及选取前k个最小元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用java写的，大概的意思就是先把两个数组的笛卡尔集放入list中，在直接排序，再利用java自带的sublist函数找出前k个

public class Solution {
    public List<int[]> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        List<int[]> ll = new ArrayList<>();

        for (int i = 0; i < nums1.length; i++) {
            for (int j = 0; j < nums2.length; j++) {
                int[] a = new int[2];
                a[0] = nums1[i];
                a[1] = nums2[j];
                ll.add(a);
            }
        }
        ll.sort(new c());
        if (k > ll.size()) {
            return ll;
        }

        return ll.subList(0, k);
    }

    static class c implements Comparator {
        @Override
        public int compare(Object o1, Object o2) {
            int[] a = (int[]) o1;
            int[] b = (int[]) o2;
            int sum1 = 0, sum2 = 0;
            for (int i = 0; i < a.length; i++) {
                sum1 += a[i];
            }
            for (int i = 0; i < b.length; i++) {
                sum2 += b[i];
            }
            return new Integer(sum1).compareTo(new Integer(sum2));
        }

    }
}