Computer >> Máy Tính >  >> Lập trình >> Python

Chương trình tìm khoảng cách tối thiểu đến phần tử đích bằng Python

Giả sử chúng ta có một số mảng và hai giá trị đích khác nhau (đích phải hiển thị bằng nums) và bắt đầu, chúng ta phải tìm một chỉ số i sao cho nums [i] =target và | i - start | là tối thiểu. Chúng ta phải trả về | i - start |.

Vì vậy, nếu đầu vào giống như nums =[3,4,5,6,7] target =7 start =2, thì đầu ra sẽ là 2 vì chỉ có một giá trị khớp với target, đó là nums [4] , do đó i =4. Bây giờ | 4-2 | =2.

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau:

  • tối thiểu:=infinity

  • đối với tôi trong phạm vi từ 0 đến kích thước của nums, hãy thực hiện

    • nếu nums [i] giống với target thì

      • nếu | i - bắt đầu |

        • tối thiểu:=| i - start |

  • trả lại tối thiểu

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

Ví dụ

from math import inf
def solve(nums, target, start):
   minimum = inf
   for i in range(len(nums)):
      if nums[i] == target:
         if abs(i - start) < minimum:
            minimum = abs(i - start)
   return minimum
nums = [3,4,5,6,7]
target = 7
start = 2
print(solve(nums, target, start))

Đầu vào

[3,4,5,6,7], 7, 2

Đầu ra

2