Submission #67462298

---

Source Code Expand

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
	"strings"
)

func min(a, b int) int {
	if a < b {
		return a
	}
	return b
}

func main() {
	reader := bufio.NewReader(os.Stdin)

	// 1行目：N
	line1, _ := reader.ReadString('\n')
	N, _ := strconv.Atoi(strings.TrimSpace(line1))

	// 2行目：h[0]～h[N-1]
	line2, _ := reader.ReadString('\n')
	hStr := strings.Fields(line2)
	h := make([]int, N)
	for i := 0; i < N; i++ {
		h[i], _ = strconv.Atoi(hStr[i])
	}

	if N == 2 {
		fmt.Println(abs(h[1] - h[0]))
		return
	}

	var prev2 int = 0                             // dp[i-2]
	var prev1 int = abs(h[1] - h[0])              // dp[i-1]

	for i := 2; i < N; i++ {
		cost1 := prev1 + abs(h[i]-h[i-1])
		cost2 := prev2 + abs(h[i]-h[i-2])
		curr := min(cost1, cost2)
		prev2 = prev1
		prev1 = curr
	}

	fmt.Println(prev1)
}

func abs(a int) int {
	if a < 0 {
		return -a
	}
	return a
}

Submission Info

Judge Result