愛麗絲和鮑勃有不同大小的糖果棒: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]
思路:分別求出兩個數組的和,分別爲sumA和sumB,重新遍歷兩個數組,由方程可知,sumA-A[m]+B[n]=sumB+A[m]-B[n],可以求出A[m]和B[n]的值,一次添加到結果數組中。
public static int[] fairCandySwap(int[] A, int[] B) {
int[] result=new int[2];
int sumA=0;
int sumB=0;
for(int i=0;i<A.length;i++){
sumA+=A[i];
}
for(int j=0;j<B.length;j++){
sumB+=B[j];
}
int average=(sumA-sumB)/2;
for(int m=0;m<A.length;m++){
for(int n=0;n<B.length;n++){
if(A[m]-average==B[n]){
result[0]=A[m];
result[1]=B[n];
break;
}
}
}
return result;
}