【每日一题】 888. 公平的糖果棒交换

【每日一题】 888. 公平的糖果棒交换

避免每日太过咸鱼,一天搞定一道LeetCode算法题

一、题目描述

难度: 简单

爱丽丝和鲍勃有不同大小的糖果棒:A[i] 是爱丽丝拥有的第 i 根糖果棒的大小,B[j] 是鲍勃拥有的第 j 根糖果棒的大小。

因为他们是朋友,所以他们想交换一根糖果棒,这样交换后,他们都有相同的糖果总量。(一个人拥有的糖果总量是他们拥有的糖果棒大小的总和。)

返回一个整数数组 ans,其中 ans[0] 是爱丽丝必须交换的糖果棒的大小,ans[1] 是 Bob 必须交换的糖果棒的大小。

如果有多个答案,你可以返回其中任何一个。保证答案存在。

提示:

  • 1 <= A.length <= 10000
  • 1 <= B.length <= 10000
  • 1 <= A[i] <= 100000
  • 1 <= B[i] <= 100000
  • 保证爱丽丝与鲍勃的糖果总量不同。
  • 答案肯定存在。

**示例 1:

输入:A = [1,1], B = [2,2]
输出:[1,2]

示例 2:

输入:A = [1,2], B = [2,3]
输出:[1,2]

二、题解

1. 解法

解题思路:哈希表

记爱丽丝的糖果棒的总大小为 sumA,鲍勃的糖果棒的总大小为sumB。设答案为 {x,y},即爱丽丝的大小为 x的糖果棒与鲍勃的大小为 y 的糖果棒交换,则有如下等式:sumA−x+y=sumB+x−y

代码

 public int[] fairCandySwap(int[] A, int[] B) {
     int[] ints = new int[2];
     int suma = Arrays.stream(A).sum();
     int sumb = Arrays.stream(B).sum();

     int delta = (suma - sumb) / 2;

     HashSet<Integer> rec = new HashSet<Integer>();
     for (int num : A) {
         rec.add(num);
     }
     for (int y : B) {
         int x = y + delta;
         if (rec.contains(x)) {
             ints[0] = x;
             ints[1] = y;
             break;
         }
     }

     return ints;
 }

复杂度分析

  • 时间复杂度:O(n+m),其中 n 是序列 A 的长度,m 是序列 B 的长度。
  • 空间复杂度:O(n)。其中 n 是序列 A 的长度。我们需要建立一个和序列 A 等大的哈希表。

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/distance-between-bus-stops

--------------最后感谢大家的阅读,愿大家技术越来越流弊!--------------

在这里插入图片描述

--------------也希望大家给我点支持,谢谢各位大佬了!!!--------------

你可能感兴趣的:(算法训练,算法,数据结构,java,leetcode,动态规划)