1643. Kth Smallest Instructions
Read the full problem statement on LeetCode.
Difficulty: hard Acceptance: 44% Topics: Array, Math, Dynamic Programming, Combinatorics
View full problem on LeetCode Reading material
Reference solution (spoiler · python)
# Time: O((m + n)^2)
# Space: O(1)
class Solution(object):
def kthSmallestPath(self, destination, k):
"""
:type destination: List[int]
:type k: int
:rtype: str
"""
def nCr(n, r): # Time: O(n), Space: O(1)
if n < r:
return 0
if n-r < r:
return nCr(n, n-r)
c = 1
for k in xrange(1, r+1):
c *= n-k+1
c //= k
return c
r, c = destination
result = []
while r+c:
count = nCr(r+(c-1), r) # the number of HX..X combinations
if k <= count: # the kth instruction is one of HX..X combinations, so go right
c -= 1
result.append('H')
else: # the kth instruction is one of VX..X combinations, so go down
k -= count # the kth one of XX..X combinations is the (k-count)th one of VX..X combinations
r -= 1
result.append('V')
return "".join(result)
Solution from kamyu104/LeetCode-Solutions · MIT