2021-03-29：無序數組arr，子數組-1和1的數量同樣多，請問最長子數組的長度是多少？

2021-03-29：無序數組arr，子數組-1和1的數量同樣多，請問最長子數組的長度是多少？

福大大 答案2021-03-29：

[1, -1, 2, 3, -4, -1, 9]變成[1, -1, 0, 0, 0, -1, 0]，累加和等於0的最長子數組長度，就是所須要的值。
求前綴和，存map。

代碼用golang編寫。代碼以下：

package main

import "fmt"

func main() {
    arr := []int{1, -1, 2, 3, -4, -1, 9}
    ret := maxLength(arr)
    fmt.Println(ret)
}

func maxLength(arr []int) int {
    if len(arr) == 0 {
        return 0
    }
    // key:前綴和
    // value : 0~value這個前綴和是最先出現key這個值的
    mmap := make(map[int]int)
    mmap[0] = -1 // important
    llen := 0
    sum := 0
    for i := 0; i < len(arr); i++ {
        if arr[i] == 1 {
            sum++
        } else if arr[i] == -1 {
            sum--
        }
        if _, ok := mmap[sum]; ok {
            llen = getMax(i-mmap[sum], llen)
        }
        if _, ok := mmap[sum]; !ok {
            mmap[sum] = i
        }
    }
    return llen
}

func getMax(a int, b int) int {
    if a > b {
        return a
    } else {
        return b
    }
}

執行結果以下：

---

評論