Leetcode 公平的糖果交换

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

 

思路:先算平均数,获得差值,以差值去比较。

 

int* fairCandySwap(int* A, int ASize, int* B, int BSize, int* returnSize){
    int *ret = malloc(sizeof(int) * 2);
    int i,j;
    int ASum = 0;
    int BSum = 0;
    int avg = 0;
    
    
    for(i = 0; i  ASize; i++)
        ASum += A[i];    
    
    
    for(i = 0; i  BSize; i++)
        BSum += B[i];  
    
    
    avg = (ASum + BSum) / 2;
    ASum = avg - ASum;
    BSum = avg - BSum;;
    
    
    for(i = 0; i  ASize; i++) {
        for(j = 0; j  BSize; j++) {

            if(A[i] + ASum == B[j]  A[i] == B[j] + BSum) {
                ret[0] = A[i];
                ret[1] = B[j];
                goto end;
            }
        }
    }

end:
    *returnSize = 2;
    return ret;
    
}

 

 

最新回复(0)
/jishuCKY9JlD1VuLMFgypi_2BK9b_2FdZP_2BrZEY8n3tlR0w_3D_3D4858227
8 简首页