LeetCode 1005. Maximize Sum Of Array After K Negations

问题

问题描述:给定一个整形数组 A,可以通过以下方式修改它:选择 i,将其反转 A[i] = -A[i],重复 K 次,可以选择相同的i反转多次。要求返回反转之后的最大和。

栗子1:

输入:A = [4, 2, 3], K = 1
输出:5
A 变为 [4, -2, 3]

栗子2:

输入: A = [3, -1, 0, 2], K = 3
输出:6
A 变为 [3, 1, 0, 2]

栗子3:

输入:A = [2, -3, -1, 5, -4], K = 2
输出:13
A 变为 [2, 3, -1, 5, 4]

注意:

  1. 1 <= A.length <= 10000
  2. 1 <= K <= 10000
  3. -100 <= A[i] <= 100

想看英文的戳这里:原题地址

解题思路

首先有一个很重要的点是,如果对一个数进行偶数次的反转,它的值保持不变。

我的解法

为了让和最大,如果有负数,需要尽量让负数进行反转,减少正数的反转。所以会分为以下几种情况来考虑:

无负数,全为正数
  • 如果 K 为偶数,那么可以保持不变,最大值就是原先数组的和。因为可以对第一个数一直反转,反转 2 次就保持原值了。

    比如 A = [4,3,2],K = 2,那么我们可以对 4 进行 2 次反转,结果不变。

  • 如果 K 为奇数,那么肯定有一个数需要变为负数。我们需要选择最小的数反转为负数,才能保证最大值。

    比如 A = [4,3,2],K = 1,那么我们可以对最小数 2 进行 1 次反转,变成[4,3,-2]
    比如 K = 5,偶数次的都可以忽略,因为对值无影响,所以我们最终计算的只是5 % 2 = 1次的反转,与上面分析一样。

有负数,有正数

在这种情况下,需要考虑K与负数个数negCount问题。

K <= negCount

如果 K <= negCount,那么很简单,排序后,只需要把前K个负数进行反转即可。
比如A = [2,-3,-1,5,-4],K = 2,反转最小的 2 个负数即可,A 变成[2,3,-1,5,4]

K > negCount

如果K > negCount,我们可以考虑将负数先全部反转,但是可能会涉及到反转正数的问题。

  • 如果 k - negCount 为偶数,那么剩余的次数落在正数头上。由于是2的倍数次反转,刚好可以抵消,正数保持不变。

    比如A = [2,-3,-1,5,-4],K = 5,由于负数有 3 个,那么刚好还剩 2 次反转,抵消,正数保持不变。

  • 如果k - negCount为奇数,则需要考虑是反转正数,还是将负数保持不变(即多余的1次落到负数头上)。

    这里需要考虑到负数的最大值正数的最小值绝对值大小问题。

    如果负数的绝对值大,那么反转正数。比如A = [2,-6,-4,5,-8],K = 4,那么 A = [-2,6,4,5,8]

    如果正数的绝对值大,那么负数保持不变。比如A = [2,-3,-1,5,-4],K = 4,那么 A = [2,3,-1,5,4]

代码如下:

// 53.66%
func largestSumAfterKNegations(_ A: [Int], _ K: Int) -> Int {
    // 存负数
    var negArray = [Int]()
    
    // 存正数
    var posArray = [Int]()
    var sum = 0
    
    for a in A {
        sum += a
        
        if a <= 0 {
            negArray.append(a)
        } else {
            posArray.append(a)
        }
    }
    
    // 排序
    negArray = negArray.sorted()
    posArray = posArray.sorted()
    
    // 需要考虑无负数的情况
    if negArray.count == 0, posArray.count > 0 {
        // 反转 2 次,抵消
        if K % 2 == 0 {
            return sum
        } else {
            return sum - 2 * posArray[0]
        }
    } else {
        // 如果 K 大于负数的个数,就需要到处理正数,优先把正数反转 2 次,即保持不变
        if K > negArray.count {
            let leftCount = K - negArray.count
            
            // count 表示需要反转负数的个数
            var count = negArray.count
            
            // 奇数,需要比较正数的最小值与负数的最大值的绝对值关系,如果是偶数,可以对负数全部操作。
            if leftCount % 2 == 1 {
                // 如果负数的绝对值小,则最后一个负数保持不变,相当于 2 次反转。
                if posArray.count > 0 {
                    if -negArray.last! < posArray[0] {
                        count -= 1
                    } else {
                        // 负数反转的增益大于正数变负数,将正数转为负数
                        sum -= 2 * posArray[0]
                    }
                } else {
                    count -= 1
                }
            }
            
            // 反转负数
            var i = 0
            while i < count {
                sum -= 2 * negArray[i]
                i += 1
            }
        } else {
            // 负数变正数
            var i = 0
            while i < K {
                sum -= 2 * negArray[i]
                i += 1
            }
        }
    }
    
    return sum
}

更简洁的解法

思路跟上面的差不多,但是没有分那么多种情况,是一个比较完整的思路。

先对整个数组进行排序,将前 K 个负数反转,最后需区分剩下的次数是奇数还是偶数。
如果是偶数,则无影响,如果是奇数,则需要减去最小值*2

// 60.98%
func largestSumAfterKNegations(_ A: [Int], _ K: Int) -> Int {
    // 先排序
    var sortedA = A.sorted()

    // 对前面的负数进行反转
    var i = 0
    var j = 0

    // 如果 K 大于负数的个数,最后需要判断是对哪个数进行反转
    while j < K, i < sortedA.count, sortedA[i] < 0 {
        sortedA[i] = -sortedA[i]
        j += 1
        i += 1
    }
    
    var sum = 0
    var k = 0

    // 记录最小的数
    var minNum = Int.max

    while k < sortedA.count {
        sum += sortedA[k]

        if sortedA[k] < minNum {
            minNum = sortedA[k]
        }

        k += 1
    }

    // 如果剩余的 K - j 是奇数,则需要减去 2*最小的数
    if (K - j) % 2 == 1 {
        sum -= 2 * minNum
    }

    return sum
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,921评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,635评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,393评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,836评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,833评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,685评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,043评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,694评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,671评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,670评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,779评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,424评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,027评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,984评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,214评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,108评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,517评论 2 343

推荐阅读更多精彩内容

  • 这是悦乐书的第376次更新,第403篇原创 01 看题和准备 今天介绍的是LeetCode算法题中Easy级别的第...
    程序员小川阅读 315评论 0 4
  • 在C语言中,五种基本数据类型存储空间长度的排列顺序是: A)char B)char=int<=float C)ch...
    夏天再来阅读 3,323评论 0 2
  • 专业考题类型管理运行工作负责人一般作业考题内容选项A选项B选项C选项D选项E选项F正确答案 变电单选GYSZ本规程...
    小白兔去钓鱼阅读 8,970评论 0 13
  • 1. 找出数组中重复的数字 题目:在一个长度为n的数组里的所有数字都在0到n-1的范围内。数组中某些数字是重复的,...
    BookThief阅读 1,744评论 0 2
  • 想到整个周末都没了,就心情不好。。。 空有一颗把蓉小馆吃完的心,可是没有一个把蓉小馆吃完的胃。
    一路李花开阅读 124评论 0 0