Leetcode 888.公平的糖果棒交换

---

一、题目描述

爱丽丝和鲍勃有不同大小的糖果棒：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
保证爱丽丝与鲍勃的糖果总量不同。
答案肯定存在。

二、分析及代码

1.公式+哈希表

记爱丽丝的糖果棒的总大小为sumA，鲍勃的糖果棒的总大小为 sumB。设所求为 {x,y}，即爱丽丝的大小为 x 的糖果棒与鲍勃的大小为 y 的糖果棒交换，则有如下等式：sumA - x + y =sumB + x - y 化简，得：x = y + (（sumA - sumB）/ 2 )。
即对于 B 中的任意一个数 y’，只要 A 中存在一个数 x’，满足 x’ = y’ + (（sumA - sumB）/ 2 )。那么{x ′ ,y ′} 即为一组可行解。
为了快速查询 A 中是否存在某个数，我们可以先将 A 中的数字存入哈希表中。然后遍历 BB 序列中的数 y’，在哈希表中查询是否有对应的 x’。

---

代码如下（示例）：

struct HashTable {
    int x;
    UT_hash_handle hh;//UT_hash_handle是头文件"uthash.h"中定义的
};

int* fairCandySwap(int* A, int ASize, int* B, int BSize, int* returnSize) {
    int sumA = 0, sumB = 0;
    for (int i = 0; i < ASize; i++) {
        sumA += A[i];//计算爱丽丝拥有的糖果棒总量
    }
    for (int i = 0; i < BSize; i++) {
        sumB += B[i];//计算鲍勃拥有的糖果棒总量
    }
    int delta = (sumA - sumB) / 2;
    struct HashTable* hashTable = NULL;
    for (int i = 0; i < ASize; i++) {
        struct HashTable* tmp;
        HASH_FIND_INT(hashTable, &A[i], tmp);//插入前先查看A[I]值是否已经在hash表里
        if (tmp == NULL) {
            tmp = malloc(sizeof(struct HashTable));
            tmp->x = A[i];
            HASH_ADD_INT(hashTable, x, tmp);//必须明确告诉插入函数，自己定义的hash结构体中键变量的名字

        }
    }
    int* ans = malloc(sizeof(int) * 2);
    for (int i = 0; i < BSize; i++) {
        int x = B[i] + delta;
        struct HashTable* tmp;
        HASH_FIND_INT(hashTable, &x, tmp);//在哈希表中查找是否存在满足条件的A[i]值
        if (tmp != NULL) {
            ans[0] = x, ans[1] = B[i];
            *returnSize = 2;
            break;
        }
    }
    return ans;
}

2.公式+排序遍历

记爱丽丝的糖果棒的总大小为sumA，鲍勃的糖果棒的总大小为 sumB。设所求为 {x,y}，即爱丽丝的大小为 x 的糖果棒与鲍勃的大小为 y 的糖果棒交换，则有如下等式：sumA - x + y =sumB + x - y 化简，得：x = y + (（sumA - sumB）/ 2 )。
即对于 B 中的任意一个数 y’，只要 A 中存在一个数 x’，满足 x’ = y’ + (（sumA - sumB）/ 2 )。那么{x ′ ,y ′} 即为一组可行解。
因为答案必然存在，我们可以将A与B按升序进行排序，利用双指针记录当前A[i]与B[j]的信息，通过与（sumA - sumB）/ 2 进行比较判断当前指针指向的一组糖果棒是否为可行解。

---

代码如下（示例）：

int cmp (const void * a,const void *b){
    return *(int *)a - *(int *)b;
}
int* fairCandySwap(int* A, int ASize, int* B, int BSize, int* returnSize){
    qsort(A,ASize,sizeof(A[0]),cmp);
    qsort(B,BSize,sizeof(B[0]),cmp);//将A，B数组按升序进行排序
    int sumA = 0,sumB = 0;
    *returnSize = 2;
    int *ans = (int*)malloc(sizeof(int) * (*returnSize));
    for(int i = 0;i < ASize;i++){
        sumA += A[i];
    }
    for(int i = 0;i < BSize;i++){
        sumB += B[i];
    }
    int diff = (sumA - sumB)/2;
    int i = 0,j = 0;
    while(i < ASize && j <BSize){
        if(A[i] - B[j] > diff){
            j++;//差值过大，此时应该增大减数的值
        }
        else if(A[i] - B[j] < diff){
            i++;//差值过小，此时应该增大被减数的值
        }
        else
        {
            ans[0] = A[i];
            ans[1] = B[j];
            break;
        }
    }
    return ans;
}

三、总结

官方认定为简单题，解题相关思路还蛮有意思的，哈希表永远滴神！