Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรมค้นหาแอมพลิจูดต่ำสุดหลังจากลบองค์ประกอบ K ใน Python


สมมติว่าเรามีตัวเลขที่เรียกว่า nums และมีค่าอื่น k หากเราลบองค์ประกอบ k ออกจาก nums ให้หาค่าต่ำสุดของ (จำนวนสูงสุด - ค่าต่ำสุดของ nums)

ดังนั้น หากอินพุตเป็น nums =[4, 10, 3, 2, 8, 9] k =3 เอาต์พุตจะเป็น 2 เพราะถ้าเราลบ 10, 8 และ 9 ค่าสูงสุดคือ 4 ค่าต่ำสุดคือ 2 ดังนั้นความแตกต่างคือ 2.

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

  • เรียงเลขรายการ

  • p :=ขนาดของ nums - k

  • m :=(องค์ประกอบสุดท้ายของ nums) - nums[0]

  • สำหรับฉันในช่วง 0 ถึงขนาดของ nums - p ทำ

    • ถ้า nums[i + p - 1] - nums[i]

      • m :=nums[i + p - 1] - nums[i]

  • กลับม

ตัวอย่าง

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น

def solve(nums, k):
   nums = sorted(nums)
   p = len(nums) - k
   m = nums[-1] - nums[0]
   for i in range(0, len(nums) - p + 1):
      if nums[i + p - 1] - nums[i] < m:
         m = nums[i + p - 1] - nums[i]
   return m

nums = [10, 4, 3, 2, 9, 8]
k = 3
print(solve(nums, k))

อินพุต

[10, 4, 3, 2, 9, 8], 3

ผลลัพธ์

2