开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第29天，点击查看活动详情

一 描述

888. 公平的糖果交换 - 力扣（LeetCode） (leetcode-cn.com)

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizes 和 bobSizes ，aliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量，bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。

两人想要互相交换一盒糖果，这样在交换之后，他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。

返回一个整数数组 answer，其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目，answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案，你可以返回其中 任何一个 。题目测试用例保证存在与输入对应的答案。

 

示例 1：

输入：aliceSizes = [1,1], bobSizes = [2,2]
输出：[1,2]

示例 2：

输入：aliceSizes = [1,2], bobSizes = [2,3]
输出：[1,2]

示例 3：

输入：aliceSizes = [2], bobSizes = [1,3]
输出：[2,3]

示例 4：

输入：aliceSizes = [1,2,5], bobSizes = [2,4]
输出：[5,4]

提示：

• 1 <= aliceSizes.length, bobSizes.length <= 10^4
• 1 <= aliceSizes[i], bobSizes[j] <= 10^5
• 爱丽丝和鲍勃的糖果总数量不同。
• 题目数据保证对于给定的输入至少存在一个有效答案。

二 分析

仔细考虑一下，因为问题肯定有解，设数组a的和为suma，b的和为sumb，假设数组a中i位置为x，b中j位置为y，恰好是解，那么就有suma-x+y==sumb-y+x，也就是suma-2x==sumb-2y,使用无序图保存每一个suma-2a[i]，在遍历数组b时，如果sumb-2b[j]出现过，那么就是解

三 答案

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& a, vector<int>& b) {
        int suma=accumulate(a.begin(),a.end(),0);
        int sumb=accumulate(b.begin(),b.end(),0);
        unordered_map<int,bool> vis;
        for(int i=0;i<a.size();i++){
            vis[suma-2*a[i]]=1;
        }
        for(int j=0;j<b.size();j++){
            if(vis[sumb-2*b[j]]){
                return {(suma-sumb+2*b[j])/2,b[j]};
            }
        }
        return {-1,-1};
    }
};