package main

import (
	"fmt"
	"sort"
)

// 740. 删除并获得点数
func deleteAndEarn(nums []int) int {
	// 1.排序
	sort.Ints(nums)
	n := len(nums)
	// 初始化dp[0]
	dp := make([][2]int, n, n)
	dp[0][0], dp[0][1] = 0, nums[0]
	for i := 1 ; i < n ; i++ {
		nv, ov := nums[i], nums[i-1]
		if nv == ov {
			dp[i][0], dp[i][1] = dp[i-1][0], dp[i-1][1] + nums[i]
		}else if nv == ov + 1 {
			dp[i][0], dp[i][1] = max(dp[i-1][0], dp[i-1][1]), dp[i-1][0] + nums[i]
		}else if nv  > ov + 1 {
			dp[i][0], dp[i][1] = max(dp[i-1][0], dp[i-1][1]), max(dp[i-1][0], dp[i-1][1]) + nums[i]
		}
	}
	return  max(dp[n-1][1], dp[n-1][0])
}
func max(a , b int) int {
	if a > b  {
		return  a
	}
	return  b
}

func main() {
	fmt.Println(deleteAndEarn([]int{2,2,3,3,3,4}))
}