DeepRest

Kth Smallest Number in Multiplication Table

Nov 16th, 2021 (edited)
67
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.58 KB | None | 0 0
  1. class Solution(object):
  2. def findKthNumber(self, m, n, k):
  3. """
  4. :type m: int
  5. :type n: int
  6. :type k: int
  7. :rtype: int
  8. """
  9. def helper(p):
  10. res = 0
  11. for i in range(1, m+1):
  12. res += min(p//i, n)
  13. return res
  14.  
  15. beg = 1
  16. end = m*n
  17. mid = (beg + end) // 2
  18. while beg <= end:
  19. if helper(mid) < k:
  20. beg = mid + 1
  21. else:
  22. end = mid - 1
  23. mid = (beg + end) // 2
  24.  
  25. return beg
Add Comment
Please, Sign In to add comment