【leetcode】公平的糖果交换 c++

题目描述：

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 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 <= 104
1 <= aliceSizes[i], bobSizes[j] <= 105
爱丽丝和鲍勃的糖果总数量不同。
题目数据保证对于给定的输入至少存在一个有效答案。

c++代码：

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& aliceSizes, vector<int>& bobSizes) {
        int n = aliceSizes.size(),m=bobSizes.size();
        int sum_alice=0,sum_bob=0; //sum_alice-x+y=sum_bob+x-y即x-y=(sum_alice-sum_bob)/2即y=x-(suma-sumb)/2
        map<int,int> b;
        for(int i=0;i<n;i++){
            sum_alice+=aliceSizes[i];
        }
        for(int i=0;i<m;i++){
            sum_bob+=bobSizes[i];
            b[bobSizes[i]]=1;
        }
        vector<int> v;
        int sum = sum_alice-sum_bob;
        if(sum%2!=0)return v; //若sum_alice-sum_bob不是2的倍数，y为非整数，不可能有解
        for(int i=0;i<n;i++){
            if(b[aliceSizes[i]-(sum/2)]==1){
                v.push_back(aliceSizes[i]);
                v.push_back(aliceSizes[i]-(sum/2));
                return v;
            }
        }
        return v;
    }
};

数学问题，列式子，表达结果。

遍历数组查找元素的问题，使用map简化时间复杂度。

总结：

找到数学关系，列数学表达式，表示结果，比多层for循环遍历快。
遍历数组查找元素，应该用map优化时间复杂度。