刷题日记-888. 公平的糖果交换

109 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述

888. 公平的糖果交换

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizesbobSizesaliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量,bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。

两人想要互相交换一盒糖果,这样在交换之后,他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。

返回一个整数数组 answer,其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目,answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案,你可以返回其中 任何一个 。题目测试用例保证存在与输入对应的答案。

 

示例 1:

输入:aliceSizes = [1,1], bobSizes = [2,2]
输出:[1,2]

示例 2:

输入:aliceSizes = [1,2], bobSizes = [2,3]
输出:[1,2]

示例 3:

输入:aliceSizes = [2], bobSizes = [1,3]
输出:[2,3]

示例 4:

输入:aliceSizes = [1,2,5], bobSizes = [2,4]
输出:[5,4]

 

提示:

  • 1 <= aliceSizes.length, bobSizes.length <= 10^4
  • 1 <= aliceSizes[i], bobSizes[j] <= 10^5
  • 爱丽丝和鲍勃的糖果总数量不同。
  • 题目数据保证对于给定的输入至少存在一个有效答案。

二、思路分析

我们设爱丽丝糖果总数目为n1,鲍勃的糖果总数目为n2

假设需要交换的糖为a和b,a属于爱丽丝,b属于鲍勃,那么

n1-a+b=n2-b+a

n1-n2=2a-2b

于是思路就为:

遍历数组A和B,分别计算总数,同时,为了下一轮遍历方便,我们将B数组存入hashmap里 遍历完成后,我们计算出n1-n2,然后遍历数组A,对每一个a算出b,然后去map里查询是否存在b,有的话,返回结果。

三、AC 代码

func fairCandySwap(A []int, B []int) []int {
	res:=[]int{0,0}
	mp:=make(map[int]int)
	sum1,sum2:=0,0
	for _,v:=range A{
		sum1+=v
	}
	for _,v:=range B{
		sum2+=v
		mp[v]++
	}
	d:=sum1-sum2
	for _,v:=range A{
		if (v*2-d)%2==0&&mp[(v*2-d)/2]!=0{
			res[0]=v
			res[1]=(v*2-d)/2
			break
		}
	}
	return res
}

范文参考

思路简单,性能高效接近100 - 公平的糖果交换 - 力扣(LeetCode) (leetcode-cn.com)