From f05d888a0b621ca4e99e2b0fb6e23c097006fe41 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Tue, 28 May 2024 10:29:13 +0200
Subject: Init

---
 progs/unparsable_programs/a771.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 progs/unparsable_programs/a771.py

(limited to 'progs/unparsable_programs/a771.py')

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
-- 
cgit v1.2.3-18-g5258