LeetCode(ID:888) 公平的糖果交换

公平的糖果交换

1. 题目描述

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

2. 示例

示例 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 = []
输出：[5,4]

3. 提示

1) 1 <= A.length <= 10000

2) 1 <= B.length <= 10000

3) 1 <= A[i] <= 100000

4) 1 <= B[i] <= 100000

5) 保丝与鲍勃的糖果总量不同。
6) 答案肯定存在。

4. 思路

1.要满足交换后两者拥有糖果重量相等，则所有糖果重量之和必然是偶数，否则没有交换方案。
2.只交换一个糖果就要达到两者拥有糖果重量相同，则存在：

$$\Sigma(A)-A[i]+B[j] = \Sigma(B)-B[j]+A[i] = avg$$
等价于
$$A[i]-B[j] = [\Sigma(A)-\Sigma(B)]/2=\delta/2;\\\delta=\Sigma(A)-\Sigma(B)$$
求解的迭代逻辑如下：

sort(A,true);//true表示升序，排序是为了加速
sort(B,true);
boolean flag = false;
for i=0:A.length-1
    for j=0:B.length
        if A[i]-B[j] < delta/2
            break;
        elseif A[i]-B[j] == delta/2
            flag = true;
            result = [A[i],B[j]];
            break;
        endif
    endfor
    if flag==true
        break;
endfor
return result;

step1：如果$A[a]-B[b]<\delta/2$，则a右移
step2：如果$A[a]-B[b]=\delta/2$，则直接输出
step3：如果$A[a]-B[b]>\delta/2$，则b右移

代码实现

public class Solution {
    public static int[] fairCandySwap(int[] A, int[] B) {
        Arrays.sort(A);
        Arrays.sort(B);
        int sumA = 0;
        int sumB = 0;
        int[] result = new int[2];
        for (int a : A) {
            sumA += a;
        }
        for (int b : B) {
            sumB += b;
        }
        double deltaDiv2 = (sumA - sumB) / 2.0;
    //  System.out.println("sumA="+Integer.toString(sumA)+",sumB="+Integer.toString(sumB)+",deltaDiv2="+Double.toString(deltaDiv2));
        boolean flag = false;
        for (int i = 0; i < A.length; i++) {
            for (int j = 0; j < B.length; j++) {
                if (A[i] - B[j] < deltaDiv2) {
                    break;
                } else if (A[i] - B[j] == deltaDiv2) {
(flag == true) {
                break;
            }
        }
        return result;
    }