diff options
| author | Santo Cariotti <santo@dcariotti.me> | 2024-05-28 10:29:13 +0200 |
|---|---|---|
| committer | Santo Cariotti <santo@dcariotti.me> | 2024-05-28 10:29:13 +0200 |
| commit | f05d888a0b621ca4e99e2b0fb6e23c097006fe41 (patch) | |
| tree | eebbb2489144112d3288393e354d19375a0aa088 /progs/unparsable_programs/a771.py | |
Init
Diffstat (limited to 'progs/unparsable_programs/a771.py')
| -rw-r--r-- | progs/unparsable_programs/a771.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/progs/unparsable_programs/a771.py b/progs/unparsable_programs/a771.py new file mode 100644 index 0000000..6343c44 --- /dev/null +++ b/progs/unparsable_programs/a771.py @@ -0,0 +1,15 @@ +import heapq
+def k_smallest_pairs(nums1, nums2, k):
+ queue = []
+ def push(i, j):
+ if i < len(nums1) and j < len(nums2):
+ heapq.heappush(queue, [nums1[i] + nums2[j], i, j])
+ push(0, 0)
+ pairs = []
+ while queue and len(pairs) < k:
+ _, i, j = heapq.heappop(queue)
+ pairs.append([nums1[i], nums2[j]])
+ push(i, j + 1)
+ if j == 0:
+ push(i + 1, 0)
+ return pairs
\ No newline at end of file |