面试题 16.21. 交换和----leetcode

给定两个整数数组，请交换一对数值（每个数组中取一个数值），使得两个数组所有元素的和相等。

返回一个数组，第一个元素是第一个数组中要交换的元素，第二个元素是第二个数组中要交换的元素。若有多个答案，返回任意一个均可。若无满足条件的数值，返回空数组。

示例:

输入: 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;
    }
}