LeetCode—公平的糖果交换（数学）

公平的糖果交换（简单）

2020年7月11日

题目来源：力扣

解题

今天比较忙，每日一题还是困难的，劝退我了，做一道简单力扣题完成打卡

这道题用数学思想就能解决

以示例一为例，A和B的总和为1+1+2+2=6，平均分给两个人，一人就是3
A想要达到3，又得拿东西出去交换，设A拿出去的为x，A收获B的为y
那就应该是A-x+y=3
B同理，应该是B-y+x=3
得到公式A-x+y=B-y+x
求解可得y=(B-A)/2+x

也就是说，如果存在这么一个y，满足x的这个条件，那就得到解了

class Solution {
    public int[] fairCandySwap(int[] A, int[] B) {
        int sum_a=0,sum_b=0;
        Set<Integer> set=new HashSet<>();
        for(int i:A){
            sum_a+=i;
        }
        for(int i:B){
            sum_b+=i;
            set.add(i);
        }
        int r=(sum_b-sum_a)/2;
        for(int i:A)
            if(set.contains(i+r)){
                return new int[]{i,i+r};
            }
        return new int[]{-1,-1};
    }
}