leetcode面试题 16.21. 交换和(二分查找)

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

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

示例:

输入: array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]
输出: [1, 3]

代码

class Solution {
    public int[] findSwapValues(int[] array1, int[] array2) {

        int sum1=0,sum2=0;
        for(int c:array1) sum1+=c;
        for(int c:array2) sum2+=c;
                int in= Math.abs(sum1-sum2);//计算数组和的差
        if(in%2!=0) return new int[]{};
        in/=2;
        Arrays.sort(array2);//先排序再二分搜索
        if(sum1<sum2)//第一个数组是需要换一个大的数字
        {
            for(int i=0;i<array1.length;i++)
            {
                int idx=Arrays.binarySearch(array2,array1[i]+in);//二分查找
                if(idx>=0) return new int[]{ array1[i],array2[idx]};//找到了符合的
            }
        }else {
            for(int i=0;i<array1.length;i++)
            {
                int idx=Arrays.binarySearch(array2,array1[i]-in);
                if(idx>=0) return new int[]{ array1[i],array2[idx]};
            }
        }
         return new int[]{};
    }
}

你可能感兴趣的:(leetcode,数据结构,leetcode,java,算法)