通过在Python中进行两个给定长度的跳转来检查是否有可能达到一个数字

假设我们在起始位置p,我们可以在d1和d2单位的任意方向(左或右)上跳跃。我们必须找到从p跳转到位置q所需的最小步数。

因此,如果输入像p = 5,q = 10,d1 = 4,d2 = 3,那么输出将是3,因为我们可以使用距离4两次向右跳转,然后可以到达位置13,然后向左跳转3个单位即可达到10个。

示例

让我们看下面的实现以更好地理解-

from math import gcd
from collections import deque
def solve(p, d1, d2, q):
   gcd_res = gcd(d1, d2)
   if (p - q) % gcd_res != 0:
      return -1
   que = deque()
   visited = set()
   que.appendleft([p, 0])
   visited.add(p)
   while len(que) > 0:
      pair = que.pop()
      point, step = pair[0], pair[1]
      if point == q:
         return step
      if point + d1 not in visited:
         que.appendleft([(point + d1), step + 1])
         visited.add(point + d1)
      if point + d2 not in visited:
         que.appendleft([(point + d2), step + 1])
         visited.add(point + d2)
      if point - d1 not in visited:
         que.appendleft([(point - d1), step + 1])
         visited.add(point - d1)
      if point - d2 not in visited:
         que.appendleft([(point - d2), step + 1])
         visited.add(point - d2)
p = 5
q = 10
d1 = 4
d2 = 3
print(solve(p, d1, d2, q))

输入值

5, 4, 3, 10
输出结果
True

猜你喜欢