Advertisement
serega1112

668 - heap

Dec 21st, 2020
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.36 KB | None | 0 0
  1. class Solution:
  2.     def findKthNumber(self, m: int, n: int, k: int) -> int:
  3.        
  4.         h = [(i, i) for i in range(1, m+1)]
  5.         heapq.heapify(h)
  6.        
  7.         while k:
  8.             val, i = heapq.heappop(h)
  9.             if val // i < n:
  10.                 heapq.heappush(h, ((val // i + 1) * i, i))
  11.             k -= 1
  12.            
  13.         return val
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement