力扣-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]

源代码

这道题目的标签就一个数组，为什么官方的题解却是哈希表呢，，我吐了。
思路：首先这道题的规律很好找，即：
返回A，B中的值的差是A，B两个数组和之差的一半。通过在草稿纸简单的举例就可以证明。
我刚开始是直接暴力解，样例通过了73/75，我吐了，只差最后几个没过，我当时就知道我的代码肯定没问题的，是这道题的解法不对，然后就发现官方题解竟然是哈希表，但是标签却只有数组。。。。。
代码如下：

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {
        vector<int> ans;
        int suma=accumulate(A.begin(),A.end(),0);
        int sumb=accumulate(B.begin(),B.end(),0);
        int count=(suma-sumb)/2;
        unordered_set<int> map(A.begin(),A.end());
        for(int j : B)
        {
            int i=j+count;
            if(map.count(i)) {
                ans.push_back(i);
                ans.push_back(j);
                break;
            }
        }
        return ans;
    }
};

复杂度分析：

• 时间复杂度：O(n+m)，其中 n 是序列 A 的长度，m 是序列 B 的长度。

• 空间复杂度：O(n)，其中 n 是序列 A 的长度。我们需要建立一个和序列 A 等大的哈希表。