力扣贪心算法--第三天

T.BS2024-04-06 2:00

前言

今天上了一天课，先写一道题吧，比较累。

内容

一、1005. K 次取反后最大化的数组和

给你一个整数数组 nums 和一个整数 k ，按以下方法修改该数组：

选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。

重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

以这种方式修改数组后，返回数组 可能的最大和 。

思路：

局部最优：让绝对值大的负数变为正数，当前数值达到最大，整体最优：整个数组和达到最大。

如果将负数都转变为正数了，K依然大于0，此时的问题是一个有序正整数序列，如何转变K次正负，让数组和达到最大。

那么又是一个贪心：局部最优：只找数值最小的正整数进行反转，当前数值和可以达到最大

第一步：将数组按照绝对值大小从大到小排序，注意要按照绝对值的大小
第二步：从前向后遍历，遇到负数将其变为正数，同时K--
第三步：如果K还大于0，那么反复转变数值最小的元素，将K用完
第四步：求和

Go 复制代码

func largestSumAfterKNegations(nums []int, k int) (sum int) {
	sort.Slice(nums, func(i, j int) bool {
		return math.Abs(float64(nums[i])) > math.Abs(float64(nums[j]))//math.Abs函数接受一个float64类型的参数，并返回一个float64类型的结果
	})

	for i := 0; i < len(nums); i++ {
		if nums[i] < 0 && k > 0 {
			nums[i] = -nums[i]
			k--
		}
	}
    if k%2==1{
        nums[len(nums)-1]=-nums[len(nums)-1]
    }
	for _, v := range nums {
		sum += v
	}
	return sum
}

Go 复制代码

func largestSumAfterKNegations(nums []int,k int)int{
   sort.Sort(byAbs(nums))
   for i:=0;i<len(nums);i++{
    if nums[i]<0&&k>0{
        nums[i]=-nums[i]
        k--
    }
   }
   if k%2==1{
    nums[len(nums)-1]=-nums[len(nums)-1]
   }
   sum:=0
   for _,v:=range nums{
    sum+=v
   }
   return sum
}

type byAbs []int
func (b byAbs)Len()int{
    return len(b)
}
func (b byAbs)Swap(i,j int){
    b[i],b[j]=b[j],b[i]
}
func (b byAbs)Less(i,j int)bool{
    return Abs(b[i])>Abs(b[j])
}
func Abs(a int)int{
    if a>0{
        return a
    }
    return -a
}

最后

...