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

Chương trình tìm độ dài của danh sách con các phần tử bất đẳng thức xen kẽ dài nhất bằng Python

Giả sử chúng ta có một danh sách các số được gọi là num và tìm độ dài của các nhà trọ trong danh sách con dài nhất sao cho quan hệ bằng nhau giữa mọi số liên tiếp thay đổi xen kẽ giữa phép toán nhỏ hơn và lớn hơn. Bất đẳng thức của hai số đầu tiên có thể không-hơn hoặc lớn hơn.

Vì vậy, nếu đầu vào là nums =[1, 2, 6, 4, 5], thì đầu ra sẽ là 4, vì danh sách con xen kẽ bất đẳng thức dài nhất là [2, 6, 4, 5] là 2 <6> 4 <5.

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

  • Định nghĩa một hàm get_direction (). Điều này sẽ mất a, b

    • trả về 0 nếu a giống b, ngược lại -1 nếu a

  • nếu kích thước của nums <2, thì

    • trả về kích thước của nums

  • max_length:=1, cur_length:=1, last_direction:=0

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

    • hướng:=get_direction (nums [i], nums [i + 1])

    • nếu hướng giống 0, thì

      • cur_length:=1

    • ngược lại khi hướng giống như last_direction thì

      • cur_length:=2

    • nếu không,

      • cur_length:=cur_length + 1

    • max_length:=tối đa của max_length và cur_length

    • last_direction:=hướng

  • trả về max_length

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

Ví dụ

class Solution:
   def solve(self, nums):
      if len(nums) < 2:
         return len(nums)
      def get_direction(a, b):
         return 0 if a == b else -1 if a < b else 1
      max_length = 1
      cur_length = 1
      last_direction = 0
      for i in range(len(nums) - 1):
         direction = get_direction(nums[i], nums[i + 1])
         if direction == 0:
            cur_length = 1
         elif direction == last_direction:
            cur_length = 2
         else:
            cur_length += 1
         max_length = max(max_length, cur_length)
         last_direction = direction
      return max_length
ob = Solution()
nums = [1, 2, 6, 4, 5]
print(ob.solve(nums))

Đầu vào

[1, 2, 6, 4, 5]

Đầu ra

4