标签:int res param public 公平 糖果 sumB sumA LeetCode.888
原题
https://leetcode-cn.com/problems/fair-candy-swap/
思路
哈希表,推导等式 sumA - x + y = sumB + x - y
解题
package com.leetcode.code;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
/**
* @ClassName Code888
* @Author ZK
* @Description
* @Date 2021/2/1 17:02
* @Version 1.0
**/
public class Code888 {
public static void main(String[] args) {
}
/**
* 暴力
* @param A
* @param B
* @return
*/
public static int[] fairCandySwap0(int[] A, int[] B) {
int[] res = new int[2];
int sumA = Arrays.stream(A).sum();
int sumB = Arrays.stream(B).sum();
for (int i = 0; i < A.length; i++) {
for (int j = 0; j < B.length; j++) {
if (sumA - A[i] + B[j] == sumB + A[i] - B[j]) {
res[0] = A[i];
res[1] = B[j];
}
}
}
return res;
}
/**
* hash表
* @param A
* @param B
* @return
*/
public static int[] fairCandySwap(int[] A, int[] B) {
int[] res = new int[2];
int sumA = Arrays.stream(A).sum();
int sumB = 0;
Set<Integer> set = new HashSet<>();
for (int x : B) {
set.add(x);
sumB += x;
}
for (int x : A) {
int temp = x - (sumA - sumB)/2;
if (set.contains(temp)) {
res[0] = x;
res[1] = temp;
}
}
return res;
}
}
标签:int,res,param,public,公平,糖果,sumB,sumA,LeetCode.888 来源: https://blog.csdn.net/wazk2008/article/details/113530545
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。