Python程式：透過檢查元素在集合中的參與度來尋找快樂

假設我們有一個包含 n 個不同整數的陣列 nums。我們還有兩個不相交的集合 A 和 B。我們有一個最初設定為 0 的快樂引數。我們遍歷 nums 中的每個整數 i。如果 i 在 A 中，則將快樂值加 1；如果 i 在 B 中，則將其減 1。我們最終需要找到最終的快樂值。

因此，如果輸入類似於 nums = [1,2,5,8,6,3] A = {5,8,9,7,3} B = {2,4,12,15}，則輸出將為 2，因為 5、8、3 在 A 中，所以快樂值為 3，但 2 在 B 中，所以將其減 1，然後快樂值為 2。

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

• 快樂值 := 0
• 對於 nums 中的每個 i，執行以下操作：
  • 如果 i 在 A 中，則
    • 快樂值 := 快樂值 + 1
  • 否則，當 i 在 B 中時，則
    • 快樂值 := 快樂值 - 1
• 返回快樂值

示例

讓我們看看以下實現以獲得更好的理解

def solve(nums, A, B):
   happiness = 0
   for i in nums:
      if i in A:
         happiness += 1
      elif i in B:
         happiness -= 1
   return happiness

nums = [1,2,5,8,6,3]
A = {5,8,9,7,3}
B = {2,4,12,15}
print(solve(nums, A, B))

輸入

[1,2,5,8,6,3], {5,8,9,7,3}, {2,4,12,15}

輸出

2

Arnab Chakraborty

更新於: 2021年10月12日

239 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

立即開始