其他分享
首页 > 其他分享> > [leetCode]888. 公平的糖果棒交换

[leetCode]888. 公平的糖果棒交换

作者:互联网

题目

在这里插入图片描述

哈希表

推导出映射关系使用哈希表求解:
假设A的总和为sumA, B的总和为sumB,交换元素x, y后A、B总和相等,即 s u m A − x + y = s u m B − y + x , 化 简 得 : x = y + ( s u m A − s u m B ) / 2 sumA - x + y = sumB - y + x,化简得: x = y + (sumA - sumB) / 2 sumA−x+y=sumB−y+x,化简得:x=y+(sumA−sumB)/2,将A中元素加入哈希表,遍历B中元素,如果符合上述映射关系,则找到了交换得一对元素。

class Solution {
    public int[] fairCandySwap(int[] A, int[] B) {
        Set<Integer> set = new HashSet<>();
        int sumA = Arrays.stream(A).sum();
        int sumB = Arrays.stream(B).sum();
        int delta = (sumA - sumB) / 2;
        for (int a : A) {
            set.add(a);
        }
        for (int b : B) {
            if (set.contains(b + delta)) {
                return new int[]{b + delta, b};
            }
        }
        return new int[0];
    }
}

标签:set,int,888,sumB,哈希,new,糖果,leetCode,sumA
来源: https://blog.csdn.net/renweiyi1487/article/details/113497146