力扣 面试题 16.21. 交换和

题目

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

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

示例

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-swap-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：哈希表

哈希表存的是arrary2中的值的 两倍。

Java实现

class Solution {
    public int[] findSwapValues(int[] array1, int[] array2) {
        int n1 = array1.length, n2 = array2.length;
        int sum1 = 0, sum2 = 0;
        Map<Integer, Integer> map = new HashMap<>();

        for (int i : array1) sum1 += i;
        for (int i : array2) {
            sum2 += i;
            map.put(2 * i, map.getOrDefault(2 * i, 0) + 1);
        }

        int diff = sum2 - sum1;

        for (int a : array1) {
            int b = diff + 2 * a;
            if (map.containsKey(b)) return new int[] {a, b / 2};
        }

        return new int[]{};
    }
}