Python程式：查詢列表中兩個數字之和等於k

假設我們有一個名為nums的數字列表，還有一個數字k，我們需要檢查列表中是否存在兩個數字加起來等於k。相同的元素不能重複使用。數字可以為負數或0。

因此，如果輸入類似於nums = [45, 18, 9, 13, 12]，k = 31，則輸出將為True，因為18 + 13 = 31

為了解決這個問題，我們將遵循以下步驟：

  • temp_set := 一個新的集合
  • 對於nums中的每個數字num，執行以下操作：
    • 如果num在temp_set中，則：
      • 返回True
    • 將(k-num)新增到temp_set中
  • 返回False

讓我們看看下面的實現，以便更好地理解：

示例

 線上演示

class Solution:
   def solve(self, nums, k):
      temp_set=set()
      for num in nums:
         if num in temp_set:
            return True
            temp_set.add(k-num)
      return False
ob = Solution()
nums = [45, 18, 9, 13, 12] k = 31 print(ob.solve(nums, k))

輸入

[45, 18, 9, 13, 12], 31

輸出

True

更新於: 2020年10月5日

755 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告