技術標籤：leetcode練習leetcode資料結構python

目錄

• • 問題描述
  • 思路分析及程式碼實現

問題描述

愛麗絲和鮑勃有不同大小的糖果棒：A[i] 是愛麗絲擁有的第 i 根糖果棒的大小，B[j] 是鮑勃擁有的第 j 根糖果棒的大小。

因為他們是朋友，所以他們想交換一根糖果棒，這樣交換後，他們都有相同的糖果總量。（一個人擁有的糖果總量是他們擁有的糖果棒大小的總和。）

返回一個整數陣列 ans，其中 ans[0] 是愛麗絲必須交換的糖果棒的大小，ans[1] 是 Bob 必須交換的糖果棒的大小。

如果有多個答案，你可以返回其中任何一個。保證答案存在。

示例 1：
輸入：A = [1,1], B = [2,2]

輸出：[1,2]
示例 2：
輸入：A = [1,2], B = [2,3]
輸出：[1,2]
示例 3：
輸入：A = [2], B = [1,3]
輸出：[2,3]
示例 4：
輸入：A = [1,2,5], B = [2,4]
輸出：[5,4]

提示：

• 1 <= A.length <= 10000
• 1 <= B.length <= 10000
• 1 <= A[i] <= 100000
• 1 <= B[i] <= 100000
• 保證愛麗絲與鮑勃的糖果總量不同。答案肯定存在。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/fair-candy-swap

思路分析及程式碼實現

我一開始是這樣做的，先求出A與平均數的差值，然後遍歷轉為集合的A，將每個元素都加上差值看看是否在轉為集合的B中，結果雖然通過了，
然時間慢的可憐

class Solution:
    def fairCandySwap(self, A: List[int], B: List[int]) -> List[int]:
        sub = (sum(A) + sum(B)) // 2 - sum(A)
        for i in set(A):
        	if i + sub in
 
 set(B):
        		return [i, i+sub]
        	
        

於是我用了雜湊表的做法，將集合A填入雜湊表中，判斷B中元素加上差值是否在雜湊表A，這樣大大提高了效率

class Solution:
    def fairCandySwap(self, A: List[int], B: List[int]) -> List[int]:
        sub = (sum(A) + sum(B)) // 2 - sum(B)
        hashdic = {}
        for i in set(A):
        	if i not in hashdic:
        		hashdic[i] = 0
        for i in set(B):
            if i + sub in hashdic.keys():
                return [i+sub,i]