leetcode—888. 公平的糖果棒交换

888. 公平的糖果棒交换

爱丽丝和鲍勃有不同大小的糖果棒：A[i] 是爱丽丝拥有的第 i 根糖果棒的大小，B[j] 是鲍勃拥有的第 j 根糖果棒的大小。

因为他们是朋友，所以他们想交换一根糖果棒，这样交换后，他们都有相同的糖果总量。（一个人拥有的糖果总量是他们拥有的糖果棒大小的总和。）

返回一个整数数组 ans，其中 ans[0] 是爱丽丝必须交换的糖果棒的大小，ans[1] 是 Bob 必须交换的糖果棒的大小。

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

示例 1：

输入：A = [1,1], B = [2,2]
输出：[1,2]

一开始的解题思路是直接暴力求解，但是很不幸，O(n^3)，超时了。

class Solution {//暴力超时
    int []ans;
    public int[] fairCandySwap(int[] A, int[] B) {
        int m = A.length;
        int n = B.length;
        int []C = new int[m];
        int []D = new int[n];
        C = A.clone();
        D = B.clone();
        ans = new int[2];
        int sum1=0,sum2=0;
        for(int i =0;i<m;i++){
            sum1 = sum1+A[i];
        }
        for(int i=0;i<n;i++){
            sum2 = sum2+B[i];
        }
        int sum = (sum1+sum2)/2;

        int tep,k=0;
        for(int i=0 ;i< m;i++){
            for(int j=0; j<n ;j++){
                tep = A[i];
                A[i] = B[j];
                B[j] = tep;
                for(int p=0; p<m; p++)
                {
                    k = k + A[p];
                }
                if(k == sum){
                ans[0]=B[j];
                ans[1]=A[i];
                return ans;
                }
                else{
                    k = 0;
                    A = C.clone();
                    B = D.clone();
                }
            }
        }
        return null;
    }
}

换个思路，两个数组的分别求和然后做差除以2，然后双重循环遍历，两个数组的数挨个做差，要是等于，就直接返回。

class Solution {
    int []ans;
    public int[] fairCandySwap(int[] A, int[] B) {
        int m = A.length;
        int n = B.length;
        ans = new int[2];
        int sum1=0,sum2=0;
        for(int i =0;i<m;i++){
            sum1 = sum1+A[i];
        }
        for(int i=0;i<n;i++){
            sum2 = sum2+B[i];
        }
        int sum = (sum2-sum1)/2;

        for(int i=0 ;i< m;i++){
            for(int j=0; j<n ;j++){
                if(B[j]-A[i] == sum){
                    ans[0]=A[i];
                    ans[1]=B[j];
                    return ans;
                }
               
                }
                }
                return null;
            }
        }