leetcode 888 Fair Candy Swap

题干

爱丽丝和鲍勃有不同大小的糖果棒：a[i]是爱丽丝第i根糖果棒的大小，b[j]是鲍勃第j根糖果棒的大小。因为他们是朋友，所以他们想每个交换一个糖果棒，这样在交换之后，他们都拥有相同的糖果总量。（一个人拥有的糖果总量是他们拥有的糖果大小之和。）返回一个整数数组ans，其中ans[0]是Alice必须交换的糖果大小，ans[1]是Bob必须交换的糖果大小。如果有多个答案，您可以返回其中任何一个。保证存在一个答案。

Example 1:

Input: A = [1,1], B = [2,2]
Output: [1,2]
Example 2:

Input: A = [1,2], B = [2,3]
Output: [1,2]
Example 3:

Input: A = [2], B = [1,3]
Output: [2,3]
Example 4:

Input: A = [1,2,5], B = [2,4]
Output: [5,4]

简单讲就是A，B数组各拿出一个数交换，确保A数组中数的和与B数组中的和相等，如果有，返回一个包含他两交换的数的数组

思路

这个要做一些简单的数学变换
设A拿出x,A原来有sumA大，B拿出y，B原来有sumB大
则sumA-x+y=sumB-y+x
即
x=y+(sumA-SumB)/2
y=x+(sumB-sumA)/2
所以当A拿出x时，只要B有 x+(sumB-sumA)/2这个数，即可返回

解法

先求出A，B的和，然后是(sumB-sumA)/2
接着，将A转换为Set,遍历A，对于每一个A中的数，只要B中含有对应的数，返回数组，转成set是因为set查询效率较高，否则超时

代码

java

class Solution {
    public int[] fairCandySwap(int[] A, int[] B) {
        int sumA=0,sumB=0;
        for(int x:A)sumA+=x;
        for(int y:B)sumB+=y;
        int mid=(sumB-sumA)/2;
        
        Set<Integer>sB=new HashSet();
        for(int y:B)sB.add(y);
        for(int x:A){
            if(sB.contains(x+mid)){
                return new int []{x,x+mid};
            }
        }
        throw null;
    }
}
    

python(最近开始学爬虫，机器学习，所以也要练习一下

class Solution(object):
    def fairCandySwap(self, A, B):
     
        sumA,sumB=sum(A),sum(B)
        #这里用//表示除出来取整，python中/表示真正的出，如3/2在python中输出1.5
        mid=(sumB-sumA)//2
        setA=set(A)
        for x in setA:
            if x+mid in B:
                return [x,x+mid]