Welcome to Subscribe On Youtube
888. Fair Candy Swap
Description
Alice and Bob have a different total number of candies. You are given two integer arrays aliceSizes
and bobSizes
where aliceSizes[i]
is the number of candies of the i^{th}
box of candy that Alice has and bobSizes[j]
is the number of candies of the j^{th}
box of candy that Bob has.
Since they are friends, they would like to exchange one candy box each so that after the exchange, they both have the same total amount of candy. The total amount of candy a person has is the sum of the number of candies in each box they have.
Return an integer array answer
where answer[0]
is the number of candies in the box that Alice must exchange, and answer[1]
is the number of candies in the box that Bob must exchange. If there are multiple answers, you may return any one of them. It is guaranteed that at least one answer exists.
Example 1:
Input: aliceSizes = [1,1], bobSizes = [2,2] Output: [1,2]
Example 2:
Input: aliceSizes = [1,2], bobSizes = [2,3] Output: [1,2]
Example 3:
Input: aliceSizes = [2], bobSizes = [1,3] Output: [2,3]
Constraints:
1 <= aliceSizes.length, bobSizes.length <= 10^{4}
1 <= aliceSizes[i], bobSizes[j] <= 10^{5}
 Alice and Bob have a different total number of candies.
 There will be at least one valid answer for the given input.
Solutions

class Solution { public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) { int s1 = 0, s2 = 0; Set<Integer> s = new HashSet<>(); for (int a : aliceSizes) { s1 += a; } for (int b : bobSizes) { s.add(b); s2 += b; } int diff = (s1  s2) >> 1; for (int a : aliceSizes) { int target = a  diff; if (s.contains(target)) { return new int[] {a, target}; } } return null; } }

class Solution { public: vector<int> fairCandySwap(vector<int>& aliceSizes, vector<int>& bobSizes) { int s1 = accumulate(aliceSizes.begin(), aliceSizes.end(), 0); int s2 = accumulate(bobSizes.begin(), bobSizes.end(), 0); int diff = (s1  s2) >> 1; unordered_set<int> s(bobSizes.begin(), bobSizes.end()); vector<int> ans; for (int& a : aliceSizes) { int target = a  diff; if (s.count(target)) { ans = vector<int>{a, target}; break; } } return ans; } };

class Solution: def fairCandySwap(self, aliceSizes: List[int], bobSizes: List[int]) > List[int]: diff = (sum(aliceSizes)  sum(bobSizes)) >> 1 s = set(bobSizes) for a in aliceSizes: target = a  diff if target in s: return [a, target]

function fairCandySwap(aliceSizes: number[], bobSizes: number[]): number[] { let s1 = aliceSizes.reduce((a, c) => a + c, 0); let s2 = bobSizes.reduce((a, c) => a + c, 0); let diff = (s1  s2) >> 1; for (let num of aliceSizes) { let target = num  diff; if (bobSizes.includes(target)) { return [num, target]; } } }