跳跃游戏(二)

package main

import (
    "fmt"
)

func main() {

    for {
        var n int 
        _, err := fmt.Scan(&n)
        if err != nil {
            break
        }
        nums := make([]int, n)
        for i := range nums {
            fmt.Scan(&nums[i])
        }
        fmt.Println(solve(nums))
    }
}

func solve(nums []int) int {
    n := len(nums)
    if n == 0 {
        return 0
    }
    canReachs := make([]bool, n)
    canReachs[n-1] = true
    r := n-1
    for i := n-2; i >= 0; i-- {
        if i+nums[i] >= r {
            r = i
            canReachs[i] = true
        }
    }
    if r > 0 {
        return -1 // 不可达
    }

    score := 0
    for i := 0; i < n;  {
        if canReachs[i] {
            score += nums[i]
        }
        k := 1
        for k := 1; k <= nums[i] && i+k<n && !canReachs[i+k]; k++ {}
        i += k
    }
    return score
}