您的位置:首页 > 健康 > 美食 > Golang | Leetcode Golang题解之第373题查找和最小的K对数字

Golang | Leetcode Golang题解之第373题查找和最小的K对数字

2025/4/19 22:18:43 来源:https://blog.csdn.net/weixin_66442839/article/details/141509549  浏览:    关键词:Golang | Leetcode Golang题解之第373题查找和最小的K对数字

题目:

题解:

func kSmallestPairs(nums1, nums2 []int, k int) (ans [][]int) {m, n := len(nums1), len(nums2)// 二分查找第 k 小的数对和left, right := nums1[0]+nums2[0], nums1[m-1]+nums2[n-1]+1pairSum := left + sort.Search(right-left, func(sum int) bool {sum += leftcnt := 0i, j := 0, n-1for i < m && j >= 0 {if nums1[i]+nums2[j] > sum {j--} else {cnt += j + 1i++}}return cnt >= k})// 找数对和小于 pairSum 的数对i := n - 1for _, num1 := range nums1 {for i >= 0 && num1+nums2[i] >= pairSum {i--}for _, num2 := range nums2[:i+1] {ans = append(ans, []int{num1, num2})if len(ans) == k {return}}}// 找数对和等于 pairSum 的数对i = n - 1for _, num1 := range nums1 {for i >= 0 && num1+nums2[i] > pairSum {i--}for j := i; j >= 0 && num1+nums2[j] == pairSum; j-- {ans = append(ans, []int{num1, nums2[j]})if len(ans) == k {return}}}return
}

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com