給定兩個整數數組,請交換一對數值(每個數組中取一個數值),使得兩個數組所有元素的和相等。
返回一個數組,第一個元素是第一個數組中要交換的元素,第二個元素是第二個數組中要交換的元素。若有多個答案,返回任意一個均可。若無滿足條件的數值,返回空數組。
示例:
輸入: array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]
輸出: [1, 3]
示例:
輸入: array1 = [1, 2, 3], array2 = [4, 5, 6]
輸出: []
提示:
1 <= array1.length, array2.length <= 100000
AC代碼:
class Solution {
public int[] findSwapValues(int[] array1, int[] array2) {
Map<Integer,Integer>map=new HashMap<>();
int flag=0,flag1=0;
for(int i=0;i<array1.length;i++){
flag+=array1[i];
}
for(int i=0;i<array2.length;i++){
flag1+=array2[i];
map.put(array2[i],1);
}
int res=flag-flag1;
int[] vis;
if(Math.abs(res)%2==1)return new int[0];
vis=new int[2];
int flag2=0;
for(int i=0;i<array1.length;i++){
int num=array1[i]-res/2;
if(map.get(num)!=null){
vis[0]=array1[i];
vis[1]=num;
flag2=1;
break;
}
}
if(flag2==0)return new int[0];
return vis;
}
}