leetcode每日一题
🚩 学如逆水行舟,不进则退。 —— 《增广贤文》
2022-12-07 1775. 通过最少操作次数使数组的和相等(middle)
题目描述:
给你两个长度可能不等的整数数组 nums1 和 nums2 。两个数组中的所有值都在 1 到 6 之间(包含 1 和 6)。
每次操作中,你可以选择 任意 数组中的任意一个整数,将它变成 1 到 6 之间 任意 的值(包含 1 和 6)。
请你返回使 nums1 中所有数的和与 nums2 中所有数的和相等的最少操作次数。如果无法使两个数组的和相等,请返回 -1 。
测试用例:
示例 1:
输入:nums1 = [1,2,3,4,5,6], nums2 = [1,1,2,2,2,2]
输出:3
解释:你可以通过 3 次操作使 nums1 中所有数的和与 nums2 中所有数的和相等。以下数组下标都从 0 开始。
- 将 nums2[0] 变为 6 。 nums1 = [1,2,3,4,5,6], nums2 = [6,1,2,2,2,2] 。
- 将 nums1[5] 变为 1 。 nums1 = [1,2,3,4,5,1], nums2 = [6,1,2,2,2,2] 。
- 将 nums1[2] 变为 2 。 nums1 = [1,2,2,4,5,1], nums2 = [6,1,2,2,2,2] 。
示例 2:
输入:nums1 = [1,1,1,1,1,1,1], nums2 = [6]
输出:-1
解释:没有办法减少 nums1 的和或者增加 nums2 的和使二者相等。
分析:
该题如果没有get到贪心的思想也许会想麻烦,但是理解到了其实是个easy题。
本题要求用最少的操作次数去操作两个数组使得它们的和相等,而每个元素都在1-6且可以变化1-6。
利用贪心的思想可将此题转换为求最小操作数使得原本更大的数组变得比另一个更小,不难发现其实其操作次数与题目要求是一样的。
代码:
func minOperations(nums1 []int, nums2 []int) int {
s1 ,s2:= getArrSum(nums1),getArrSum(nums2)
n :=s1-s2
//如果一开始满足直接return 0
if s1==s2{
return 0
}
if s1<s2{ //默认s1>s2
return minOperations(nums2,nums1)
}
//计算s1中每个元素最大可变小值与s2中每个元素最大可变大值存储进arr
arr:=make([]int,0)
for _,v :=range nums1{
arr=append(arr,v-1)
}
for _,v:=range nums2{
arr=append(arr,6-v)
}
//将arr从大到小排序,通过贪心的思想,每次操作得到最大改变,直到s2>s1就是最小操作次数
sort.Slice(arr,func(i,j int)bool{
return arr[i]>=arr[j]
})
cnt:=0
for _,v :=range arr{
cnt+=1
n-=v
if n<=0{
return cnt
}
}
return -1
}
func getArrSum(arr []int )int{
sum:=0
for _,v:=range arr{
sum+=v
}
return sum
}