力扣888.公平的糖果棒交换

来源：力扣（LeetCode） 链接：

https://leetcode-cn.com/problems/fair-candy-swap

爱丽丝和鲍勃有不同大小的糖果棒：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]

解法一：双重循环

先求出两个人的糖果棒总量即数组A和数组B中数据之和，利用双层循环将A中的每个数与B中的每个数都进行交换，交换之后比较看两个数组中数据之和是否相等，若相等则返回刚刚交换的两个数据，否则返回NULL。

 int Sum(int *ar,int len)//求数组中的数据之和
 {
     int sum=0;
     for(int i=0;i<len;i++)
     {
          sum+=ar[i];
     }
     return sum;
 }
int* fairCandySwap(int* A, int ASize, int* B, int BSize, int* returnSize){
    *returnSize=2;
    int suma=Sum(A,ASize);//爱丽丝的糖果总量
    int sumb=Sum(B,BSize);//鲍勃的糖果总量
    if(suma==sumb) return NULL;//如果二者的糖果总量一开始就相等直接返回NULL
    int *ans=(int *)malloc(sizeof(int)*2);
    for(int i=0;i<ASize;i++)
    {
        for(int j=0;j<BSize;j++)
        {
            if(suma-A[i]+B[j]==sumb+A[i]-B[j])//让A中的每个数据和B中的每个数据都做交换，交换完成后看两个数组中的数据之后是否相等
            {
                ans[0]=A[i];
                ans[1]=B[j];
                return ans;
            }
        }
    }
    return NULL;
}

解法二：双指针

先算出两个数组中数据的总和，然后再利用STL中的vector对两个数组排序，由于suma-sumb的值是两个人糖果总量的差值，因此我们需要在两个数组中查找二者之差为（suma-sumb）/2的两个元素并且返回。

vector<int> fairCandySwap(vector<int>& A, vector<int>& B)
	{
		int sumA = 0;
		int sumB = 0;
		for (auto& x : A)
		{
			sumA += x;
		}
		for (auto& x : B)
		{
			sumB += x;
		}
		int target = (sumA - sumB) / 2;
		sort(A.begin(), A.end());
		sort(B.begin(), B.end());
		int i = 0;
		int j = 0;
		while (i < A.size() && j < B.size())
		{
			int curr = A[i] - B[j];
			if (curr == target)
			{
				return { A[i], B[j] };
			}
			else if (curr > target)
			{
				++j;
			}
			else
			{
				++i;
			}
		}
		return vector<int>();
	}