leetcode: 373. 查找和最小的K对数字(中等,堆)

题目:

在这里插入图片描述

分析:

看完题就感觉做过,
但是还是没想出来,

看到求前k个的最值,那么一般都是用到了堆。

洛谷上的方法:

在这里插入图片描述

差点走入一个误区,注意,各行是没有重复的。

这么长时间:vector s = { t.first,t.second.first };这都不知道!!!

代码:

class Solution {
public:
    struct cmp {
		bool operator() (const pair<int, pair<int, int> >& a, const pair<int, pair<int, int> >& b) {
			return a.first + a.second.first > b.first + b.second.first;
		}
	};
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        int len1=nums1.size();
        int len2=nums2.size(); 
        k = min(k,len1*len2);
        vector<vector<int> > vv;
        if(k==0) return vv;
        priority_queue <pair<int,pair<int,int> >,vector<pair<int, pair<int, int> > >, cmp > q;
        for (int i = 0; i < nums1.size(); i++) {
			pair<int, pair<int, int> > t(nums1[i], pair<int, int>(nums2[0] , 0));
			q.push(t);
		}
        for(int i=0;i<k;i++)
        {
            pair<int, pair<int, int> > pp = q.top();
            q.pop();
            vector<int> s = { pp.first,pp.second.first };
            vv.push_back(s);
            if(pp.second.second<nums2.size()-1)
            {
                pp.second.second++;
                pp.second.first=nums2[pp.second.second];
                q.push(pp);
            }
        }
    return vv;
    }
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
可以用 Kotlin 实现一个小根来解决这个问题。首先,我们可以将 nums1 中的每个数与 nums2 中的第一个数组合成一对,并将其加入小根中。然后,我们每次从中取出和最小的数对 (u, v),并将 nums1 中和 u 相邻的数和 nums2 中和 v 相邻的数组成的数对加入中。重复这个步骤 k 次即可得到和最小的 k 个数对。 以下是 Kotlin 实现的代码: ```kotlin import java.util.* fun kSmallestPairs(nums1: IntArray, nums2: IntArray, k: Int): List<List<Int>> { val heap = PriorityQueue<List<Int>>(compareBy { it[0] + it[1] }) for (i in nums1.indices) { heap.offer(listOf(nums1[i], nums2[0], 0)) } val result = mutableListOf<List<Int>>() for (i in 0 until k) { val pair = heap.poll() ?: break result.add(listOf(pair[0], pair[1])) if (pair[2] == nums2.size - 1) continue heap.offer(listOf(pair[0], nums2[pair[2] + 1], pair[2] + 1)) } return result } ``` 这个函数接受三个参数:两个整数数组 nums1 和 nums2,以及一个整数 k。它返回一个包含最小的 k 个数对的列表。在函数内部,我们首先创建一个小根 heap,并将 nums1 中的每个数与 nums2 中的第一个数组合成一对并加入中。然后,我们循环 k 次,每次从中取出和最小的数对 (u, v),将其加入结果列表 result 中,并将 nums1 中和 u 相邻的数和 nums2 中和 v 相邻的数组成的数对加入中。最后,我们返回结果列表 result。 参考资料: - LeetCode 题目 373. 查找最小的 K 对数字 - Kotlin 标准库中的 PriorityQueue 类

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值